python训练好的模型保存与加载

将训练好的模型保存下来，避免重复训练：
以下使用pickle和joblib保存训练好的模型以及模型的加载

模型训练，以svm为例子

from sklearn import datasets
clf =  svm.SVC()
iris = datasets.load_iris()
x, y = iris.data, iris.target
clf.fit(x, y)

pickle保存模型

import pickle
s = pickle.dumps(clf)
f = open('svm.txt', 'w')
f.write(s)
f.close()

pickle加载模型

import pickle
f = open('svm.txt','r')
content = f.read()
clf = pickle.loads(content)

joblib保存模型

from sklearn.externals import joblib
joblib.dump(clf,'svm.pkl')

joblib加载模型

from sklearn.externals import joblib
clf=joblib.load('svm.pkl')