时间戳

首页 > 解决方案 > sklearn 混淆矩阵在错误的位置显示错误的尺寸/刻度线

python - sklearn 混淆矩阵在错误的位置显示错误的尺寸/刻度线

问题描述

我正在尝试显示一个混淆矩阵，但我一生都无法弄清楚为什么它拒绝以适当的方式显示。这是我的代码：

import numpy as np
import itertools
from sklearn.metrics import confusion_matrix

def plot_confusion_matrix(cm, classes,
                          normalize=False,
                          title='Confusion matrix',
                          cmap=plt.cm.winter):
    if normalize:
        cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis]
    plt.imshow(cm, interpolation='nearest', cmap=cmap)
    plt.title(title, fontsize=30)
    plt.colorbar()
    tick_marks = np.arange(len(classes))
    plt.xticks(tick_marks, classes, fontsize=20)
    plt.yticks(tick_marks, classes, fontsize=20)

    fmt = '.2f' if normalize else 'd'
    thresh = cm.max() / 2.

    for i, j in itertools.product(range(cm.shape[0]), range(cm.shape[1])):
        plt.text(j, i, format(cm[i, j], fmt), horizontalalignment="center", 
                 color="white" if cm[i, j] < thresh else "black", fontsize=40)

    plt.tight_layout()
    plt.ylabel('True label', fontsize=30)
    plt.xlabel('Predicted label', fontsize=30)

    return plt

cm = confusion_matrix(y_test, y_predicted_counts)
fig = plt.figure(figsize=(10, 10))
plot = plot_confusion_matrix(cm, classes=['Unsure','No','Yes'], normalize=False, title='Confusion matrix')
plt.show()
print(cm)

这就是显示的内容：

错误的混淆矩阵

任何帮助，将不胜感激。提前致谢。

标签： pythonscikit-learnconfusion-matrix

解决方案

对于imshow您需要指定的调用origin='lower'（默认为'upper'；他们可能在某个时候更改了此设置，并且 scikit-learn 文档没有更新他们的示例）。因此，以下应该可以解决问题：

plt.imshow(cm, interpolation='nearest', cmap=cmap, origin='lower')
#                                                    ^
#                                                    |
# added origin='lower'  ------------------------------

推荐阅读