机器学习基础 互动版

保存训练过的模型

要把训练好的模型保存到硬盘上,可以使用joblib函数。

from sklearn import svm
from sklearn import datasets
from sklearn.externals import joblib
digits = datasets.load_digits()
clf = svm.SVC(gamma=0.001, C=100.)                #使用支持向量机进行分类
clf.fit(digits.data[:-4], digits.target[:-4])    #将除最后4组的数据输入进行训练
joblib.dump(clf,"digits.pkl")                    #将模型保存到硬盘上,文件名digits.pkl
clf1=joblib.load("digits.pkl")                    #从硬盘上加载模型
clf1.predict(digits.data[-4:])                    #用保存的模型预测最后4组的数据

可以看出预测结果没有改变。此时硬盘上会出现digits.pkl文件和若干后缀名为npy的文件,这些都是模型保存的文件。