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