40
правок
Изменения
XGBoost
,→Пример использования с помощью библиотеки xgboost
param = {
'max_depth': 3, # максимальная глубина дерева 'eta': 0.3, # параметр обучения
'silent': 1,
'objective': 'multi:softprob', # функция потерь 'num_class': 3} # число классов num_round = 20 # число итераций
bst = xgb.train(param, dtrain, num_round)