11 保存模型
来源:互联网 发布:淘宝花椒官方旗舰店 编辑:程序博客网 时间:2024/05/01 09:32
保存模型
- 保存模型
- pickle
- joblib
总算到了最后一次的课程了,我们训练好了一个Model 以后总需要保存和再次预测, 所以保存和读取我们的sklearn model也是同样重要的一步。sklearn官方文档主要介绍两种保存Model的模块pickle与joblib。
1 pickle
首先简单建立与训练一个SVCModel。
from sklearn import svmfrom sklearn import datasetsclf = svm.SVC()iris = datasets.load_iris()X, y = iris.data, iris.targetclf.fit(X,y)
使用pickle来保存与读取训练好的Model。 (若忘记什么是pickle,可以回顾13.8 pickle 保存数据视频。)
import pickle #pickle模块#保存Model(注:save文件夹要预先建立,否则会报错)with open('save/clf.pickle', 'wb') as f: pickle.dump(clf, f)#读取Modelwith open('save/clf.pickle', 'rb') as f: clf2 = pickle.load(f) #测试读取后的Model print(clf2.predict(X[0:1]))# [0]
2 joblib
joblib是sklearn的外部模块。
from sklearn.externals import joblib #jbolib模块#保存Model(注:save文件夹要预先建立,否则会报错)joblib.dump(clf, 'save/clf.pkl')#读取Modelclf3 = joblib.load('save/clf.pkl')#测试读取后的Modelprint(clf3.predict(X[0:1]))# [0]
最后可以知道joblib在使用上比较容易,读取速度也相对pickle快。
阅读全文
1 0
- 11 保存模型
- Tensorflow:模型保存/模型恢复?
- 如何保存Keras模型
- 模型保存save
- Tensorflow进行保存模型
- tensorflow 模型保存问
- 如何保存Keras模型?
- 如何保存Keras模型
- 模型保存(持久化)
- keras如何保存模型
- TensorFlow保存读取模型
- 模型的保存
- 如何保存keras模型
- keras 保存模型和加载模型
- keras 保存模型和加载模型
- tensorflow之保存模型与加载模型
- EMF模型保存时加密
- 保存相关模型数据(HABTM)
- valgrind小例
- mouseenter、mouseleave和mouseout、mouseover的区别
- eclipse servlet简单的用户登录
- java的系统学习之路(三)
- Guide for camera usage in DB410c(1)
- 11 保存模型
- Java中的编译时类型与运行时类型详谈
- LeetCode@String_412_Fizz_Buzz
- 军事中的兵法
- Druid简介(Spring Boot + Mybatis + Druid数据源【自己定制】)
- 初识PCA
- 使用Gitblit来搭建Git服务器
- Eric4无法生成compile form的问题
- C语言:If 语句实现的分支语句