save 保存自己的模型
pickle
用的是pickle的形势得到一个pickle的文件
存储
from sklearn import svm
from sklearn import datasets
import pickle
clf = svm.SVC()
iris = datasets.load_iris()
X,y = iris.data,iris.target
clf.fit(X,y)
with open('save/clf.pickle','wb') as f:
pickle.dump(clf,f)
取出进行预测
利用pickle文件取出，得到的clf2
根据第一行的数据预测出花的种类是0
from sklearn import svm
from sklearn import datasets
import pickle
iris = datasets.load_iris()
X,y = iris.data,iris.target
with open('save/clf.pickle','rb') as f:
clf2 = pickle.load(f)
print(clf2.predict(X[0:1]))
#[0]
利用joblib
joblib其中会更快速，利用了多线程的技术
在文件的存储上也可以看出差异
from sklearn import svm
from sklearn import datasets
from sklearn.externals import joblib
import pickle
clf = svm.SVC()
iris = datasets.load_iris()
X,y = iris.data,iris.target
clf.fit(X,y)
joblib.dump(clf,'save/clf.pkl')
clf3 = joblib.load('save/clf.pkl')
print(clf3.predict(X[0:1]))
#[0]
