模型保存与加载

python内置持久性模型pickle

import pickle
from sklearn import svm
from sklearn import datasets
clf = svm.SVC()
X, y = datasets.load_iris(return_X_y=True)
clf.fit(X, y)
s = pickle.dumps(clf)
clf2 = pickle.loads(s)
clf2.predict(X[0:1])

joblib替代pickle

joblib在大数据上效率更高，但只能在磁盘上机型处理

from joblib import dump, load
# 保存模型`在这里插入代码片`
dump(clf,'filename.joblib')
# 加载模型
clf = load('filename.joblib')

注意：dump,load也接受类似文件的对象。