要加載保存的模型,可以使用joblib
模塊中的load
和dump
方法。下面是一個示例代碼:
from sklearn.externals import joblib
# 保存模型
model = ... # 使用Scikit-learn訓練得到的模型
joblib.dump(model, 'model.pkl')
# 加載模型
loaded_model = joblib.load('model.pkl')
在上面的代碼中,首先使用joblib.dump
保存訓練得到的模型到model.pkl
文件中。然后使用joblib.load
加載保存的模型,得到loaded_model
對象,即可繼續(xù)使用該模型進行預測等操作。