xgboost.train 与 XGBClassifier

Question

我正在使用 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")

Answer 1

XGBClassifier是一个scikit-learn兼容的类，可以与其他 scikit-learn 实用程序结合使用。

除此之外，它只是 的包装器xgb.train，您不需要在其中提供高级对象，例如Booster等。

只需将您的数据发送到等fit()，predict()内部就会自动转换为适当的对象。