XGBoost, XGBClassifier: неизвестная ошибка целевой функции

Я пытаюсь обучить XGBClassifier, но получаю эту ошибку. Я использую xgboost версии 1.1.0. Я использовал pip install xgboost для установки xgboost, и я также обновил его.

param_dict = {'n_estimators':i, 'max_depth':j, 'objective':'binary:logistic'}

clf = xgb.XGBClassifier(param_dict)
clf.fit(X_tr_1, y_train)


XGBoostError: [08:00:25] C:\Users\Administrator\workspace\xgboost-win64_release_1.1.0\src\objective\objective.cc:26: Unknown objective function: `{'objective': 'binary:logistic', 'eta': 0.02, 'max_depth': 4}`
Objective candidate: survival:aft
Objective candidate: binary:hinge
Objective candidate: multi:softprob
Objective candidate: multi:softmax
Objective candidate: rank:ndcg
Objective candidate: rank:map
Objective candidate: rank:pairwise
Objective candidate: reg:squaredlogerror
Objective candidate: reg:logistic
Objective candidate: binary:logistic
Objective candidate: reg:gamma
Objective candidate: reg:tweedie
Objective candidate: count:poisson
Objective candidate: survival:cox
Objective candidate: binary:logitraw
Objective candidate: reg:linear
Objective candidate: reg:squarederror

1 ответ

Предполагая, что вы, должно быть, использовали технику GridSearch для определения лучших гиперпараметров или даже явно указали их, Правильный способ передать объект словаря param_dict в качестве аргумента метода классификатора XGBoost -

clf = xgb.XGBClassifier(**param_dict)
Другие вопросы по тегам