sklearn模型保存

使用sklearn训练完模型之后，只有将模型持久化到硬盘上，才能方便下次直接使用。

第一种方式：使用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)
SVC(C=1.0, cache_size=200, class_weight=None, coef0=0.0,decision_function_shape='ovr', degree=3, gamma='auto', kernel='rbf',max_iter=-1, probability=False, random_state=None, shrinking=True,tol=0.001, verbose=False)>>> import pickle
>>> s = pickle.dumps(clf)
>>> clf2 = pickle.loads(s)
>>> clf2.predict(X[0:1])
array([0])
>>> y[0]
0

第二种方式：使用sklearn.external.joblib

>>> from sklearn.externals import joblib
>>> joblib.dump(clf, 'filename.pkl')
...
加载时
>>> clf = joblib.load('filename.pkl')

预测模型标记语言：PMML

PMML利用XML描述和存储数据挖掘模型，是一个已经被W3C所接受的标准。
它以纯文本形式存储模型数据，因此跨平台、跨语言性能很好。当模型部署时，往往需要使用此种语言。
PMML是一种标准，许多机器学习库都支持将模型导出为PMML。

决策树导出为图像数据

sklearn.tree包中可以将决策树导出为dot类型的文件，dot类型的文件是一种描述图的纯文本文件，它可以使用文本编辑器打开查看，里面是一堆定义结点和边的命令。

from sklearn import tree, datasetsdata = datasets.load_iris()
x = data['data']
y = data['target']
clf = tree.DecisionTreeClassifier()
clf.fit(x, y)
tree.export_graphviz(clf, out_file="haha.dot")

参考资料

model persistence
PMML
知乎讨论
关于DOT文件格式

转载于:https://www.cnblogs.com/weiyinfu/p/7887850.html

sklearn模型保存相关推荐

Sklearn 模型保存和使用
sklearn中,训练的模型需要保存,可使用joblib库. 模型 model= PCA(n_componenets = 2) model=.fit(x) 使用joblib库保存 import job ...
Sklearn，TensorFlow，keras模型保存与读取
一.sklearn模型保存与读取 1.保存 1 from sklearn.externals import joblib 2 from sklearn import svm 3 X = [[0, 0 ...
python手动将机器学习模型保存为json文件
python手动将机器学习模型保存为json文件 # 导入需要的包和库: # Import Required packages #-------------------------# Import t ...
TensorFlow模型保存和加载方法
TensorFlow模型保存和加载方法模型保存 import tensorflow as tfw1 = tf.Variable(tf.constant(2.0, shape=[1]), name=& ...
【Python】Python语言学习：pip工具使用知识，模型保存pickle，PDF与docx相互转换处理...
2022年第01周. 这一周的Python语言学习,记录如下. 01 pip工具使用知识 1 pip是什么? pip是一个用Python写的用于安装和管理包的包管理系统.它连接一个叫做Python P ...
【Python】pip工具使用知识，模型保存pickle，PDF与docx相互转换处理
2022年第01周. 这一周的Python语言学习,记录如下. 01 pip工具使用知识 1 pip是什么? pip是一个用Python写的用于安装和管理包的包管理系统.它连接一个叫做Python P ...
python打包exe 之打包sklearn模型中的各种坑及其解决方法。
之前学习了如何打包,如何建立虚拟环境打包,以及如何带资源打包exe. python打包成exe 可执行文件 .教程使用pipenv建立虚拟环境解决python打包exe文件过大的问题(附打包带图标, ...
python实现sklearn的基本操作流程，sklearn预处理方法，sklearn基础算法的使用，以及sklearn模型的选择方法。
python实现sklearn的基本操作流程,sklearn预处理方法,sklearn基础算法的使用,以及sklearn模型的选择方法. 一.数据的获取与分析 1.读取数据 2.分析数据二.数据的预 ...
python机器学习手册——学习代码（导入、处理、时间、图像、降维、sklearn模型、神经网络）
一.向量.矩阵.数组 # 1.1 创建向量 array #一维数组 import numpy as np vector_row = np.array([1,2,3]) # 创建行向量 vector_c ...