SimpleImputer 对象没有属性 _fit_dtype

Question

我有一个训练有素的scikit-learn模型管道（包括一个SimpleImputer ），我正试图将其投入生产。 但是，在生产环境中运行时出现以下错误。

SimpleImputer object has no attribute _fit_dtype

我该如何解决这个问题？

Answer 1

这是在开发和生产环境中使用不同版本的 scikit-learn 的结果。 该模型已使用一个版本进行训练，然后用于另一个版本。

这可以通过使用以下方法将开发环境中的当前库版本存储在requirements.txt文件中来解决：

pip list --format=freeze > requirements.txt

在生产环境中，安装相同的库版本：

pip install -r requirements.txt