微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何从xgboost中的最佳迭代保存模型?

如何解决如何从xgboost中的最佳迭代保存模型?

我正在使用XGBClassifier进行图像分类。因为我是机器学习和xgboost的新手。但是最近我知道,经过一定迭代后,我使用pickle库保存的模型是最后一次迭代,而不是最佳迭代。谁能告诉我如何从最佳迭代中保存模型?显然我正在使用早停。 如果我在提问时有任何错误,我深表歉意。请我尽快解决该问题,因为我的论文需要它。 那些向我建议较旧问题以寻求最佳迭代的人,请问我的问题有所不同,我想将最佳迭代以pickle格式保存,以便将来我可以使用它,而不仅仅是在同一代码中用于以后的预测。 谢谢。

解决方法

使用joblib dump / load保存/加载模型,并获得模型的增强器,以获得最佳迭代

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。