需要 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)电话会给你想要的。