XGBoost 基础教程

XGBoost 笔记

xgboost 如何将训练的模型进行保存和再次加载

XGBoost 笔记 XGBoost 笔记


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')
...