sklearn保存模型

# View more python tutorials on my Youtube and Youku channel!!!

# Youtube video tutorial: https://www.youtube.com/channel/UCdyjiB5H8Pu7aDTNVXTTpcg

# Youku video tutorial: http://i.youku.com/pythontutorial

"""

Please note, this code is only for python 3+. If you are using python 2+, please modify the code accordingly.

"""

from __future__ import print_function

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)

# method 1: pickle

import pickle

# save

with open('save/clf.pickle', 'wb') as f:

    pickle.dump(clf, f)

# restore

with open('save/clf.pickle', 'rb') as f:

   clf2 = pickle.load(f)

   print(clf2.predict(X[0:1]))

# method 2: joblib

from sklearn.externals import joblib

# Save

joblib.dump(clf, 'save/clf.pkl')

# restore

clf3 = joblib.load('save/clf.pkl')

print(clf3.predict(X[0:1]))

最后可以知道joblib在使用上比较容易，读取速度也相对pickle快。