码迷,mamicode.com
首页 > 其他好文 > 详细

SKlearn - 模型保存

时间:2020-07-01 09:47:33      阅读:111      评论:0      收藏:0      [点我收藏+]

标签:简单   roc   orm   ext   学习   job   extern   保存   als   

sklearn 中 模型保存有两种方式,版本不同,可查看官网,这里只做简单记录

 

方式一:序列化

序列化 的具体用法请自行研究

>>> from sklearn import svm
>>> from sklearn import datasets
>>> clf = svm.SVC()
>>> X, y= datasets.load_iris(return_X_y=True)
>>> clf.fit(X, y)
SVC()

>>> import pickle
>>> s = pickle.dumps(clf)
>>> clf2 = pickle.loads(s)
>>> clf2.predict(X[0:1])
array([0])
>>> y[0]
0

 

方式二:joblib  【推荐】

可以保存 机器学习 模型,也可以保存 数据标准化 等 方法;

import numpy as np
from joblib import dump, load
# from sklearn.externals import joblib    # 不同版本的用法
from sklearn.preprocessing import StandardScaler

clf = 333
dump(clf, 3.joblib)       # 保存
clf = load(3.joblib)      # 加载
print(clf)


### 示例:保存 数据标准化 模型
x = np.array([[2, 4, 3]]).T
std = StandardScaler().fit(x)
x_std = std.transform(x)
print(x_std)
# [[-1.22474487]
#  [ 1.22474487]
#  [ 0.        ]]

dump(std, std.joblib)
std_store = load(std.joblib)

x2 = np.array([[4, 2, 3]]).T
print(std_store.transform(x2))
# [[ 1.22474487]
#  [-1.22474487]
#  [ 0.        ]]

 

 

参考资料:

https://scikit-learn.org/stable/modules/model_persistence.html  官网

https://www.cnblogs.com/zichun-zeng/p/4761602.html

SKlearn - 模型保存

标签:简单   roc   orm   ext   学习   job   extern   保存   als   

原文地址:https://www.cnblogs.com/yanshw/p/13217237.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!