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