pickle and joblib

在訓練模型後將模型保存的方法，以避免下次重複訓練。
使用pickle模塊或者sklearn內部的joblib
1、使用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)

import pickle
s=pickle.dumps(clf)
f=open('svm.txt','w')
f.write(s)
f.close()
f2=open('svm.txt','r')
s2=f2.read()
clf2=pickle.loads(s2)
clf2.score(X,y)
2、使用joblib
joblib更適合大數據量的模型，且只能往硬盤存儲，不能往字符串存儲
from sklearn.externals import joblib
joblib.dump(clf,'filename.pkl')
clf=joblib.load('filename.pkl')
---------------------
做者：ahora_wzy
來源：CSDN
原文：https://blog.csdn.net/ahora_wzy/article/details/51188453
版權聲明：本文爲博主原創文章，轉載請附上博文連接！