XGBoost 基础教程

XGBoost 笔记

original icon
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://www.knowledgedict.com/tutorial/xgboost-save-and-load.html

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