要加载保存的模型,可以使用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
对象,即可继续使用该模型进行预测等操作。