python xgboost继续培训现有模型

Question

假设我构建了一个xgboost模型：

bst = xgb.train(param0, dtrain1, num_round, evals=[(dtrain, "training")])

其中：

• param0是xgb的一组参数，
• dtrain1是准备接受培训的DMatrix
• num_round是轮数

然后，我将模型保存到磁盘：

bst.save_model("xgbmodel")

稍后，我想重新加载我保存的模型，继续使用dtrain2进行培训

有没有人知道该怎么做？

Answer 1

您甚至不必从磁盘加载模型并重新训练。

您需要做的就是使用附加参数的相同xgb.train命令：xgb_model =（xgboost模型完整路径名称，您可以像问题或Booster对象一样保存）。

示例：

bst = xgb.train(param0, dtrain2, num_round, evals=[(dtrain, "training")], xgb_model='xgbmodel')

祝你好运！