温馨提示×

Scikit-learn中怎么加载保存的模型

小亿
103
2024-05-10 18:21:53
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

要加载保存的模型,可以使用joblib模块中的loaddump方法。下面是一个示例代码:

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

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读:Scikit-learn中怎么部署模型

0