from sklearn.externals import joblib以及dump存储模型,joblib.load()调用原模型

书scikit-learn机器学习-page51示例代码:

import matplotlib.pyplot as plt
from sklearn import datasets
digits=datasets.load_digits()

from sklearn.model_selection import train_test_split
Xtrain, Xtest, Ytrain, Ytest = train_test_split(digits.data, digits.target, test_size=0.2, random_state=2)

from sklearn import svm
clf=svm.SVC(gamma=0.001, C=100.)
clf.fit(Xtrain, Ytrain)
clf.score(Xtest, Ytest)

Ypred=clf.predict(Xtest)
#查看预测的情况
fig, axes=plt.subplots(4,4,figsize=(8,8))
for i, ax in enumerate(axes.flat):
ax.imshow(Xtest[i].reshape(8,8),cmap=plt.cm.gray_r,interpolation=‘nearest’)
ax.text(0.05,0.05,str(Ypred[i]),fontsize=32, transform=ax.transAxes, color=‘green’ if Ypred[i]== Ytest[i] else ‘red’)
ax.text(0,8, 0.05, str(Ytest[i]), fontsize=32, transform=ax.transAxes, color=‘black’)
ax.set_xticks([])
ax.set_yticks([])
#保存模型参数
from sklearn.externals import joblib
joblib.dump(clf, ‘digits_svm.pkl’)
clf=joblib.load(‘digits_svm.pkl’)
Ypred=clf.predict(Xtest)
clf.score(Ytest, Ypred)
#之前的support vector 模型就是clf, 使用sklearn包中的externals 中的joblib.dump功能将模型储存并且命名为digits_svm.pkl
#届时使用joblib.load调用原模型即可

在这里插入图片描述

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值