机器学习模型保存之joblib、pckle报错问题

在使用sklearn训练大规模机器学习模型后,尝试使用joblib和pickle进行保存和加载时遇到问题。joblib在加载大型模型(如34.7G的knn模型)时报错,而pickle则因大小限制(不超过4GB)无法保存大型模型。目前寻求解决方案。
摘要由CSDN通过智能技术生成

机器学习模型保存之joblib、pckle报错问题

遇到的问题

使用sklearn训练机器学习模型时,为了能够使得训练好的模型可以多次使用,我们一般都会将模型存储到本地,下一次需要用的时候直接加载就可以了,不用重新进行漫长的训练过程。
但最近在实际操作中,我们经常使用的joblib和pickle均报错,特此在这里进行记录,也希望能有人提出好的解决方法

joblib

正常情况下,joblib保存模型非常方便,只需要两行代码:

from sklearn.externals import joblib
joblib.dump(clf,"trained_model.m")

clf就是我们已经训练好的机器学习模型,trained_model.m则是我们为模型起的名字,不添加其他路径的情况下,它会保存在和脚本文件相同的文件夹下面。
当我们需要加载模型时,只需要使用:


                
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值