一、模型保存和调用【保存方法不一样,调用方法也不一样】
joblib
import joblib
#保存
joblib.dump(model, "model1.pkl")
#加载
model = joblib.load(open('model1.pkl', 'rb'))
pre = model.predict(X_test)
pickle
import pickle
#保存
pickle.dump(model, open('model1.pkl', 'wb'))
# 加载
model = pickle.load(open('model1.pkl', 'rb'))
Y_pred = model.predict(X_test)
两种不同的保存方式都可将模型保存成pkl文件,在调用的时候要根据保存的不同方法选择对应的调用方法,不然就会报错:UnpicklingError: invalid load key, '\x00'. invalid load key, ‘\x10‘.