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

 

本节整理了 XGBoost 开发中的一些技巧及经验之谈。 ...
xgboost 加载模型时,出现 AssertError:read can not have position excceed buffer ...
XGBoost 是一个 GBDT 即梯度提升算法的开源实现软件库,是 eXtreme Gradient Boosting 的简写,表示 XG ...
用 python 机器学习库训练测试样本时,往往会用到经过特征处理之后的 libsvm 格式的样本数据,这里列出了加载读取 libsvm 数 ...
xgboost booster 参数表示选择哪种 boost 分类器,它总共有两大种选择(决策树与线性模型),三个具体选择。 ...