机器学习之保存与加载.pickle模型文件

import pickle
from sklearn.externals import joblib
from sklearn.svm import SVC
from sklearn import datasets#定义一个分类器
svm = SVC()iris = datasets.load_iris()
X = iris.data
y = iris.target#训练模型
svm.fit(X,y)#1.保存成Python支持的文件格式Pickle
#在当前目录下可以看到svm.pickle
with open('svm.pickle','wb') as fw:  pickle.dump(svm,fw)
#加载svm.pickle
with open('svm.pickle','rb') as fr:new_svm1 = pickle.load(fr)# print (new_svm1.predict(X[0:1]))#2.保存成sklearn自带的文件格式Joblib
joblib.dump(svm,'svm.pkl')
#加载svm.pkl
new_svm2 = joblib.load('svm.pkl')
print (new_svm2.predict(X[0:1]))

总结：

1.训练好一个Model以后需要保存和再次预测
2.有两个模块用来保存模型 : pickle和joblib
3.Sklearn的模型导出本质上是利用Python的Pickle机制。对Python的函数进行序列化，也就是把训练好的Transformer函数序列化并存为文件。代码流程：1.保存Model(注:save文件夹要预先建立，否则会报错)joblib.dump(clf, ‘save/clf.pkl’)2.读取Modelclf2 = joblib.load(‘save/clf.pkl’)3.测试读取后的Modelprint(clf2.predict(X[0:1]))**

转载于:https://www.cnblogs.com/ywjfx/p/11152696.html

机器学习之保存与加载.pickle模型文件相关推荐

JavaScript玩转机器学习：保存并加载 tf.Model
JavaScript玩转机器学习:保存并加载 tf.Model 保存并加载 tf.Model TensorFlow.js提供了保存和加载模型的功能,这些模型可以是使用LayersAPI创建的或从现有T ...
java加载pmml模型文件报错_PMML总结与思考PMML模型生成和加载示例
在机器学习用于产品的时候,我们经常会遇到跨平台的问题.比如我们用Python基于一系列的机器学习库训练了一个模型,但是有时候其他的产品和项目想把这个模型集成进去,但是这些产品很多只支持某些特定的生产环 ...
xBIM 实战01 在浏览器中加载IFC模型文件
系列目录 [已更新最新开发文章,点击查看详细] 一.创建Web项目打开VS,新建Web项目,选择 .NET Framework 4.5 选择一个空的项目新建完成后,项目结构如下: 二.添 ...
Keras如何保存、加载Keras模型
链接 Keras中文文档一.如何保存 Keras 模型? 1.保存/加载整个模型(结构 + 权重 + 优化器状态) 不建议使用 pickle 或 cPickle 来保存 Keras 模型. 你可以使 ...
python保存与加载LGBM模型，并解决报错TypeError: Need at least one training dataset or model file or model string..
方法一(适用于sklearn接口:from lightgbm import LGBMClassifier) 比如现在有一个lgbm_model import joblib# 保存 joblib.dum ...
动态加载Fbx模型文件
为什么80%的码农都做不了架构师?>>> 动态加载FBX文件方法1(已测试过) 1 将模型拖动到场景中 ,调整好位置.(制作prefab需要) 2 新建Resources(如 ...
vue使用three.js加载.FBX模型文件
1.需要安装的依赖 //three依赖yarn add three//tween.js依赖yarn add @tweenjs/tween.js 2. 封装组件 draw.vue 放在component ...
pytorch基础知识整理（三）模型保存与加载
1, torch.save(); troch.load() torch.save()使用python的pickle模块把目标保存到磁盘,可以用来保存模型.张量.字典等,文件后缀名一般用pth或pt或p ...
gensim bm25模型保存与加载
gensim bm25模型保存与加载 1. 模型保存 2. 模型加载 20210719修改: python version:3.6.12 gensim version:3.8.3 使用bm25模型计算 ...
tensor和模型保存与加载 PyTorch
PyTorch教程-7:PyTorch中保存与加载tensor和模型详解保存和读取Tensor PyTorch中的tensor可以保存成 .pt 或者 .pth 格式的文件,使用torch.save ...

机器学习之保存与加载.pickle模型文件

机器学习之保存与加载.pickle模型文件相关推荐

最新文章

热门文章