小编Max*_*Max的帖子

我正在使用 python 以增量方式（逐块）拟合 xgboost 模型。我遇到了一个使用 xgboost.train 的解决方案，但我不知道如何处理它返回的 Booster 对象。例如，XGBClassifier 有 fit、predict、predict_proba 等选项。

这是我正在一点一点地读取数据的 for 循环内部发生的事情：

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')
modelXG.save_model("xgbmodel")

如果我们使用“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')