保存sklearn模型管道时,pickle的替代方法

时间:2018-06-01 01:11:18

标签: python machine-learning scikit-learn

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

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

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

0 个答案:

没有答案