如何获取sklearn.metrics.classification_report的输出作为dict？

Question

我一直试图以字典的形式获得分类报告。 因此，根据scikit-learn 0.20文档，我这样做：

from sklearn import metrics

rep = metrics.classification_report(y_true, y_pred, output_dict=True)

但是出现错误提示

TypeError: classification_report() got an unexpected keyword argument 'output_dict'
The scikit-learn module in my machine was initially 0.19.1 but even after updating it to 0.20, the same error message shows.

Answer 1

只要安装了scikit-learn 0.20.0，就不会显示此错误。如果您在Jupyter笔记本电脑中尝试此操作，请确保使用以下方式在笔记本电脑中反映正确的版本：

import sklearn
print(sklearn.__version__)

如果您已经升级了scikit-learn，但是jupyter显示的软件包版本错误，请确保在当前环境中安装了jupyter（并在新终端中重新启动jupyter）。