python - RuntimeError : The reset parameter is False but there is no n_features_in_ attribute. 是否安装了这个估算器？

Question

我正在使用 pickle 保存 sklearn.impute.SimpleImputer 对象。输入器适合

imputer = SimpleImputer(missing_values = np.nan, strategy = 'mean')
imputer = imputer.fit(train)

然后我用

pickle.dump(imputer,open('imputer.pkl','wb'))

并使用在 flask 应用程序中加载模型

imputer = pickle.load(open('imputer.pkl','rb'))

如果我检查 imputer.statistics_ ，我可以成功地看到应该估算的值。然而，当我跑

imputer.transform(test)

在我的 Flask 应用程序中使用加载了 pickle 的输入器，我收到以下错误:
“重置参数为 False 但没有”
运行时错误:重置参数为 False 但没有 n_features_in_ 属性。这个估计器安装了吗？
我正在使用 sklearn 0.23.1 版来拟合输入器并转换数据。有没有人对此有任何见解？请让我知道我可以提供哪些其他信息。

Answer 1

将 sklearn 版本恢复到 0.22.2.post1 可以解决您的问题:

pip install scikit-learn==0.22.2.post1

它对我有用。