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