本文主要介绍scikit-learn中的模型的保存与加载。
import pickle
from sklearn.svm import SVC
from sklearn import datasets
svm = SVC()
iris = datasets.load_iris()
X = iris.data
y = iris.target
svm.fit(X, y)
with open('svm.pickle', 'wb') as fw:
pickle.dump(svm, fw)
with open('svm.pickle', 'rb') as fr:
new_svm = pickle.load(fr)
print new_svm.predict(X[0:1])
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
[0]
from sklearn.externals import joblib
from sklearn.svm import SVC
from sklearn import datasets
svm = SVC()
iris = datasets.load_iris()
X = iris.data
y = iris.target
svm.fit(X, y)
joblib.dump(svm, 'svm.pkl')
new_svm = joblib.load('svm.pkl')
print new_svm.predict(X[0:1])
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
[0]
转载自:http://blog.csdn.net/quincuntial/article/details/71244324
参考:
http://scikit-learn.org/stable/modules/model_persistence.html
https://docs.python.org/2/library/pickle.html#
http://sklearn.lzjqsdd.com/tutorial/basic/tutorial.html#model-persistence