xgboost 如何将训练的模型进行保存和再次加载?
保存模型
xgboost 训练的模型其实是 Booster
对象(多棵弱分类器组成的强分类器),它提供了多个模型保存的函数,常用 save_model
函数,具体示例如下:
import xgboost as xgb
...
# 训练
model = xgb.train(xgb_params, d_train)
# 保存
model.save_model('model.xgb')
...
加载模型
可以通过 Booster 的构造函数加载模型,示例如下:
import xgboost as xgb
...
model = xgb.Booster(model_file='model.xgb')
...