xgboost.需要训练概率输出

Question

XGBClassifier输出概率，如果我们使用方法"predict_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对象。如果您习惯了 sklearn 模型的.predict()方法，则在分类问题的情况下，xgb.Booster.predict()调用返回概率而不是预期的标签。所以modelXG.predict(dtest)电话会给你你想要你需要的。