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