保存sklearn模型的最佳方法是什么？

Question

我正在开发一个python桌面应用程序.这个应用程序做了一些预测.现在我使用python脚本训练我的sklearn模型,将模型的参数保存为yaml文件中的字典.然后,我将这个yaml构建到我的python应用程序中.然后,当我使用应用程序时,使用字典中的参数重新创建模型.我意识到,拥有不同版本的sklearn的人会收到错误.我试图将我的模型保存在pickle文件中,但在这种情况下,当app在具有不同版本sklearn的机器上运行时,它会产生一些警告.

Answer 1

无法保证给定的 sklearn 模型在 sklearn 版本之间兼容。事实上，版本之间的实现或内部 API 可能会发生变化。请在此处查看更多信息。

如果您考虑一个版本，最好的方法确实是 pickle，而不是将参数保存在 yaml 文件中。最好使用 joblib 来执行此操作。请在此处查看更多信息。