一、模型保存和调用【保存方法不一样,调用方法也不一样】


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‘.
 
 

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐