需要xgboost.train概率输出

Question

如果我们使用方法“predict_proba”，XGBClassifier会输出概率，但是，当我使用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模型的xgb.Booster.predict()方法，则.predict()调用会在分类问题而非预期标签的情况下返回概率。因此modelXG.predict(dtest)来电会让您满意。