如何存储训练模型

1.使用pickle

#保存Model(注:save文件夹要预先建立，否则会报错)
with open('save/clf2.pickle', 'wb') as f:
    pickle.dump(model, f)
#加载Model
with open('save/clf.pickle', 'rb') as f:
    model = pickle.load(f)

2.使用joblib

from sklearn.externals import joblib
# 保存模型
joblib.dump(lr,  "./ML/test.pkl")   # lr是训练好的模型， "./ML/test.pkl"是模型要保存的路径及保存模型的文件名，其中，'pkl' 是sklearn中默认的保存格式gai
lr = joblib.load("./ML/test.pkl")
# 进行模型的预测
y_pred = lr.predict(x_test)  # 加载出来的模型跟我们训练出来的模型一样，有相同的参数