Требуется вывод вероятности xgboost.train

Question

XGBClassifier выводит вероятности, если мы используем метод "gnast_Proba ", однако, когда я обучаю модель с использованием xgboost.train, я не могу понять, как получить вероятности в качестве выходных данных. Вот кусок моего кода:

dtrain=xgb.DMatrix(X_train, label=y)
param = {'max_depth':2, 'eta':1, 'silent':1, 'objective':'binary:logistic'}
modelXG=xgb.train(param,dtrain,xgb_model='xgbmodel')

Answer 1

xgboost.train() возвращает объект xgb.Booster. Вызов xgb.Booster.predict() возвращает вероятности в случае проблемы классификации вместо ожидаемых меток, если вы используете методы .predict() моделей sklearn. Так что modelXG.predict(dtest) вызов даст вам, что вам нужно.