Scikit学习API xgboost允许在线培训?

时间:2017-07-17 22:41:34

标签: python machine-learning scikit-learn xgboost

根据API,似乎正常的xgboost接口允许此选项:

xgboost.train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None, maximize=False, early_stopping_rounds=None, evals_result=None, verbose_eval=True, xgb_model=None, callbacks=None, learning_rates=None)

在此选项中,可以输入xgb_model以允许继续在同一模型上进行培训。

但是,我正在使用xgboost的scikit学习API,因此我可以将分类器放在scikit管道中,以及其他很好的工具,例如随机搜索超参数调整。

有没有人知道任何(虽然是hacky)允许xgboost的scikitlearn api在线培训的方式?

1 个答案:

答案 0 :(得分:0)

我不认为sklearn包装器可以逐步训练模型。使用 warm_start 参数可以在某种程度上实现这一壮举。但是,用于XGBoost的sklearn包装器没有该参数。 因此,如果您想参加增量培训,则可能必须切换到xgboost的官方API版本。