Sklearn, ImportError : top_k_accuracy_score

Question

我在导入任何其他指标（例如 accuracy_score、classification_report 时没有遇到任何问题) 但 top_k_accuracy_score 会触发导入错误。我运行我在 Google Colab 上的代码。

代码片段：

from sklearn.metrics import accuracy_score, classification_report, top_k_accuracy_score

print(top_k_accuracy_score(cat_y_test, preds, k = 3))

错误信息：

ImportError: cannot import name 'top_k_accuracy_score' from 'sklearn.metrics' (/usr/local/lib/python3.7/dist-packages/sklearn/metrics/__init__.py)

我在运行他们在 docs 中提供的示例时也遇到了同样的错误。

提前致谢。

Answer 1

我在使用 scikit-learn 0.21.2 时遇到了同样的错误。

尝试通过调用

将您的 scikit-learn 更新到最新版本（2021 年 8 月为 0.24.2）

pip install --upgrade scikit-learn

或者在 google collab 上使用此命令：

!pip install --upgrade scikit-learn

它解决了我的错误。

python