遇到的问题
使用sklearn训练机器学习模型时,为了能够使得训练好的模型可以多次使用,我们一般都会将模型存储到本地,下一次需要用的时候直接加载就可以了,不用重新进行漫长的训练过程。
但最近在实际操作中,我们经常使用的joblib和pickle均报错,特此在这里进行记录,也希望能有人提出好的解决方法。
joblib
正常情况下,joblib保存模型非常方便,只需要两行代码:
from sklearn.externals import joblib
joblib.dump(clf,"trained_model.m")
clf就是我们已经训练好的机器学习模型,trained_model.m则是我们为模型起的名字,不添加其他路径的情况下,它会保存在和脚本文件相同的文件夹下面。
当我们需要加载模型时,只需要使用: