python 绘制混淆矩（ confusion matrix）

迷南。 2023-01-21 11:21 19阅读 0赞

# 代码 #

import matplotlib.pyplot as plt
    import numpy as np
    
    classes = ['ang', 'hap', 'neu', 'sad']#标签列表
    confusion_matrix = np.array(([91, 1, 4, 2], [6, 92, 2, 2], [2, 3, 92, 3], [8, 13, 4, 90]))#二维混淆矩阵
    
    plt.imshow(confusion_matrix, interpolation='nearest', cmap=plt.cm.Oranges)  # 按照像素显示出矩阵
    plt.title('confusion_matrix')
    plt.colorbar()
    tick_marks = np.arange(len(classes))
    plt.xticks(tick_marks, classes, rotation=0)  # 倾斜
    plt.yticks(tick_marks, classes)
    
    thresh = confusion_matrix.max() / 2.
    # iters = [[i,j] for i in range(len(classes)) for j in range((classes))]
    # ij配对，遍历矩阵迭代器
    iters = np.reshape([[[i, j] for j in range(4)] for i in range(4)], (confusion_matrix.size, 2))
    for i, j in iters:
        plt.text(j, i, format(confusion_matrix[i, j]), va='center', ha='center')  # 显示对应的数字
    
    plt.ylabel('Real label')
    plt.xlabel('Prediction')
    plt.tight_layout()
    # plt.show()
    plt.savefig('confusion_matrix2.png', format='png')

# 结果展示 #

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2MDc5OTg2_size_16_color_FFFFFF_t_70_pic_center]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2MDc5OTg2_size_16_color_FFFFFF_t_70_pic_center]: /images/20221020/108349112adb4645b63342e778b550fe.png

发表评论取消回复

表情：

评论列表（有 0 条评论，19人围观）

还没有评论，来说两句吧...

相关阅读

相关几何不变矩--Hu矩

【图像算法】图像特征： ---------------------------------------------------------------------------

た入场券/ 2023年08月17日 16:15/ 0 赞/ 145 阅读

相关混淆矩阵tf.math.confusion_matrix

![20191009191333910.png][] [日萌社][Link 1] [人工智能AI：Keras PyTorch MXNet TensorFlow Pa

矫情吗；*/ 2023年07月03日 02:41/ 0 赞/ 13 阅读

相关 python 绘制混淆矩（ confusion matrix）

代码 import matplotlib.pyplot as plt import numpy as np classes = ['an

迷南。/ 2023年01月21日 11:21/ 0 赞/ 20 阅读

相关分类模型confusion matrix混淆矩阵可视化

之前写过一篇关于在scikit-learn工具包中，可视化estimator分类模型分类结果的confusion matrix混淆矩阵可视化的方法，具体可以参考

约定不等于承诺〃/ 2022年09月09日 01:53/ 0 赞/ 247 阅读

相关一阶矩、二阶矩含义

矩有一阶矩、二阶矩、以后统称高阶矩，最常用的有一阶和二阶矩。一阶矩又叫静矩，是对函数与自变量的积xf(x)的积分(连续函数)或求和(离散函数)。力学中用以表示f(x)分

￡神魔★判官ぃ/ 2022年08月09日 06:27/ 0 赞/ 450 阅读

相关 opencv中的图像矩（空间矩，中心矩，归一化中心矩，Hu矩）

1. [ opencv计算图像矩的API][opencv_API] C++:Moments moments (InputArray array, bool binaryIma

拼搏现实的明天。/ 2022年06月02日 10:25/ 0 赞/ 254 阅读

相关 confusion matrix 混淆矩阵绘制

原理横轴一般是predict label，纵轴是ground truth label，对角线是预测正确的概率或个数绘制 sklearn 中confusion\

淡淡的烟草味﹌/ 2022年05月12日 02:50/ 0 赞/ 315 阅读

相关 Confused Codes

inner_net = False p_ip = '1.1.1.1' t_ip = '2.2.2.2' def switch

水深无声/ 2022年04月22日 06:06/ 0 赞/ 172 阅读

相关 Confusion Matrix（混淆矩阵）

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ub

偏执的太偏执、/ 2022年03月01日 18:26/ 0 赞/ 301 阅读

相关关于confusion_matrix()返回的矩阵的索引顺序（类别顺序）

转载至：[https://blog.csdn.net/m0\_38061927/article/details/77198990][https_blog.csdn.net_m0

淩亂°似流年/ 2021年11月01日 05:10/ 0 赞/ 319 阅读