2017-10-05 65 views
4

我工作的一个Python桌面应用程序。这个应用程序做了一些预测。现在我使用python脚本训练我sklearn模型,该模型的参数保存在YAML文件的字典。然后,我将这个yaml构建到我的python应用程序中。然后,当我使用应用程序时,将使用字典中的参数重新创建模型。我意识到,拥有不同版本sklearn的人会遇到错误。我想救我的模型处于困境之中的文件,但在这种情况下,当产生了应用程序的机器上运行不同版本sklearn的一些警告。保存sklearn模型的最佳方法是什么?

回答

4

不保证给定sklearn模型将sklearn版本之间的兼容。事实上,实现或内部API可能在版本之间改变。查看更多信息here

如果你考虑一个版本,最好的办法确实是泡菜,而不是参数保存在YAML文件中。使用joblib更好。查看更多信息here

4

我意识到,拥有不同版本sklearn的人会得到一个错误。

在这种情况下,使用隔离创建的Python环境virtualenvs

相关问题