Изменения

Перейти к: навигация, поиск

CatBoost

4 байта убрано, 18:56, 25 ноября 2018
Пример использования
== Пример использования ==
* Делим данные на тренировочное и тестовое множество
X_train, X_validation, y_train, y_validation = train_test_split(X, y, train_size=0.5, random_state=1234)
print(X_train.shape, X_validation.shape)
* Создаем классификатор
from catboost import CatBoostClassifier
)
* Обучаемся
best_model.fit(
X_train, y_train,
)
* Вывод числа деревьев в модели
print ('Resulting tree count:', best_model.tree_count_)
''Resulting tree count: 217''
Анонимный участник

Навигация