sklearn学习笔记(一)——模型保存(pickle模块和joblib模块)
2016-04-19 10:57
375 查看
在训练模型后将模型保存的方法,以免下次重复训练。
使用pickle模块或者sklearn内部的joblib
一、使用pickle模块
from sklearn import svm
from sklearn import datasets
clf=svm.SVC()
iris=datasets.load_iris()
X,y=iris.data,iris.target
clf.fit(X,y)
import pickle
s=pickle.dumps(clf)
f=open('svm.txt','w')
f.write(s)
f.close()
f2=open('svm.txt','r')
s2=f2.read()
clf2=pickle.loads(s2)
clf2.score(X,y)
二、使用joblib
joblib更适合大数据量的模型,且只能往硬盘存储,不能往字符串存储
from sklearn.externals import joblib
joblib.dump(clf,'filename.pkl')
clf=joblib.load('filename.pkl')
使用pickle模块或者sklearn内部的joblib
一、使用pickle模块
from sklearn import svm
from sklearn import datasets
clf=svm.SVC()
iris=datasets.load_iris()
X,y=iris.data,iris.target
clf.fit(X,y)
import pickle
s=pickle.dumps(clf)
f=open('svm.txt','w')
f.write(s)
f.close()
f2=open('svm.txt','r')
s2=f2.read()
clf2=pickle.loads(s2)
clf2.score(X,y)
二、使用joblib
joblib更适合大数据量的模型,且只能往硬盘存储,不能往字符串存储
from sklearn.externals import joblib
joblib.dump(clf,'filename.pkl')
clf=joblib.load('filename.pkl')
相关文章推荐
- 保证分布式系统数据一致性的6种方案
- ml的线性回归应用(python语言)
- ztree树 叶子节点路径的集合
- stm32 IAR环境printf函数调用
- 深入剖析tomcat读书笔记——连接器
- Sencha Cmd 6 和 Ext JS 6 指南文档(部分官方文档中文翻译)
- Android studio code template个性化设置
- (LeetCode 342) Power of Four
- HDU 1879 继续畅通工程
- Win7_64Bit安装Office2007注册表1402错误的解决办法
- GT9xx系列------gesture mode在framework里的实现
- HTML小技巧
- Javascript中匿名函数的多种调用方式
- ECMAscript6 map
- Storm例子
- JavaSE 扫雷项目展示
- 潮汕“七样羹”,吃了变后生
- continue、break、return的区别
- MyEclipse取消自动跳到Console窗口
- Unity中注意事项 其实很简单,应该是自己粗心吧- -