如何保存sklearn训练好的算法模型

作者:炼己者

本博客所有内容以学习、研究和分享为主,如需转载,请联系本人,标明作者和出处,并且是非商业用途,谢谢!


用sklearn训练好我们的算法模型,以后肯定还会用到,所以我们要对它们进行保存和读取。方法有很多,我就讲一种,用joblib进行模型的读写

# 训练模型
clf = svm.SVC(probability=True)
clf.fit(x,y)
# 给保存的模型的名字加上时间标签,以区分训练过程中产生的不同的模型
mdhms = time.strftime('%d%H%M', time.localtime(time.time()))
# 保存的模型的文件名
file = r'C:\Users\zhangyh4\Desktop\xietong1012\model\svm.joblib' + '_' + mdhms
# 保存模型
joblib.dump(clf,file)
# 读取模型
svm_model = joblib.load(file)

以上便是模型的保存和读取,大家有疑问可以在底下评论交流

你可能感兴趣的:(如何保存sklearn训练好的算法模型)