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

方法一

import pickle
#save model
save_file = open('save_path','w')
pickle.dump('你的模型',save_file)
save_file.close()
#load model
model_file = open('save_path','r')
model = pickle.load(model_file)
model_file.close()

方法一可能会出现报错

方法二

from sklearn.externals import joblib
#如果这行不对就
import joblib
#save model
save_file = 'save_path'
joblib.dump('你的模型',save_file)
#load model
save_file ='save_path'
model = joblib.load(save_file)