为什么我的孩子sklearn.metrics公司矩阵输出是否换位？

Question

我的理解是混淆矩阵应该在列中显示真实的类，在行中显示预测的类。因此，列的总和应等于真系列的值\u counts（）。你知道吗

我在这里举了一个例子：

from sklearn.metrics import confusion_matrix

pred = [0, 0, 0, 1]
true = [1, 1, 1, 1]

confusion_matrix(true, pred)

为什么这会给我以下输出？当然应该是这个的转置？你知道吗

array([[0, 0],
       [3, 1]], dtype=int64)

Answer 1

scikit-learn's confusion matrix遵循特定的顺序和结构。你知道吗

引用：https://scikit-learn.org/stable/auto_examples/model_selection/plot_confusion_matrix.html#sphx-glr-auto-examples-model-selection-plot-confusion-matrix-py

Answer 2

这种混乱可能是因为sklearn遵循了与维基百科文章不同的混乱轴矩阵约定。所以，回答你的问题：它以特定的格式给你输出，因为sklearn希望你以特定的方式阅读它。你知道吗

以下是编写混淆矩阵的两种不同方法：

• sklearn's way of reading/writing confusion matrix：行中的真标签和列中的预测标签
• wikipedia example与sklearn相反