0
点赞
收藏
分享

微信扫一扫

【机器学习】保存和载入模型 -sklearn

慕容冲_a4b8 2022-03-17 阅读 76

文章目录


一、模型的保存和载入

模型的保存和载入方便我们将训练好的模型保存在本地或发送在网上,载入模型方便我们在不同的环境下进行测试。 使用 pickle 可以进行保存与载入,也可以使用sklearn自带的函数:

import sklearn.externals as sk_externals
sk_externals.joblib.dump(model,'model.pickle') #保存
model = sk_externals.joblib.load('model.pickle') #载入

参考链接

  1. 十分钟上手sklearn:特征提取,常用模型,交叉验证
举报

相关推荐

0 条评论