需要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()methods，那么.predict()调用会在分类问题而不是预期标签的情况下返回概率。所以modelXG.predict(dtest)电话会给你想要的。