首页 > 解决方案 > 保存 sklearn 模型管道时的 pickle 替代品

问题描述

我有一个 scikit-learn 管道(最小-最大缩放 -> poly(degree=2) -> lasso 模型)。我希望能够在不进行酸洗的情况下保存和使用这个模型,主要是因为 sklearn 版本之间的兼容性,并且在某种程度上,对泡菜的安全漏洞感到不安。

如果我只是使用套索,我可以简单地保存和重用模型的系数,但不太确定如何轻松保存/重用模型管道的第一部分。

我可以编写代码来重新生成前两个步骤,但理想情况下有一种更简单的方法来完成它。任何人都可以提出替代方法来保存和重复使用完整的 sklearn 管道吗?有什么好的例子吗?

标签: pythonmachine-learningscikit-learn

解决方案


推荐阅读