数学建模学习(68):机器学习训练模型的保存与模型使用

网友投稿 664 2022-05-29

机器学习模型通常需要数小时或数天才能运行,尤其是在具有许多特征的大型数据集上。如果你的机器坏了,你会丢失你的模型,你需要从头开始重新训练它。

Pickle 是一个有用的 Python 工具,可让你保存模型,最大限度地减少冗长的重新训练,并允许你共享、提交和重新加载预先训练的机器学习模型。

Pickle 是一个通用的对象序列化模块,可用于序列化和反序列化对象。虽然它最常与保存和重新加载经过训练的机器学习模型相关联,但它实际上可以用于任何类型的对象。以下是如何使用 Pickle 将训练好的模型保存到文件并重新加载以获取预测。

数学建模学习(68):机器学习训练模型的保存与模型使用

模型保存

接着上篇内容:数学建模学习67,代码也是在原来的基础上继续添加,两行即可保存模型,这里把model保存为model.pkl:

import pickle pickle.dump(model, open('model.pkl'

1

2

机器学习

版权声明:本文内容由网络用户投稿,版权归原作者所有,本站不拥有其著作权,亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容,请联系我们jiasou666@gmail.com 处理,核实后本网站将在24小时内删除侵权内容。

上一篇:PyTorch实现正则化和Dropout
下一篇:在可插拔数据库上如何监控进程内存的使用
相关文章