混淆矩阵可视化 python实现

最新推荐文章于 2024-02-07 22:26:04 发布

修行小仙女

最新推荐文章于 2024-02-07 22:26:04 发布

阅读量2k

点赞数 1

分类专栏： python数据操作文章标签：数据可视化 python 可视化列表

本文链接：https://blog.youkuaiyun.com/weiyunna/article/details/112782653

版权

本文展示了如何使用Python的matplotlib库创建混淆矩阵的可视化。通过显示一个10x10的矩阵，每个单元格对应预测类别和真实类别的组合，矩阵中的数值表示在相应类别上的预测次数。代码中设置了矩阵的标签，包括旅行、运动、聚餐等，并添加了颜色条以增强视觉效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

if __name__ == '__main__':
    import matplotlib.pyplot as plt
    import numpy as np

    confusion = np.array(([95,0,0,0,0,0, 5,0, 0,0], [17.5,80,0,0,2.5,0,0,0,0,0], [10, 2.5, 85,0,0,0,0,2.5,0,0],
    [10,12.5,5,57.5,0,2.5,0,2.5,5,5],[2.5,5,0,2.5,65,25,0,0,0,0],[7.5,5,2.5,0,17.5,57.5,0,0,5,5],
    [27.5,7.5,2.5,0,0,0,62.5,0,0,0],[0,15,0,10,0,0,0,70,2.5,2.5],[2.5,0,0,0,0,0,0,0,97.5,0],
                          [0,2.5,0,2.5,2.5,0,0,2.5,2.5,87.5]))//混淆矩阵
    # 热度图，后面是指定的颜色块，可设置其他的不同颜色
    plt.imshow(confusion, cmap=plt.cm.Reds)
    # ticks 坐标轴的坐标点
    # label 坐标轴标签说明
    indices = range(len(confusion))
    # 第一个是迭代对象，表示坐标的显示顺序，第二个参数是坐标轴显示列表
    # plt.xticks(indices, [0, 1, 2])
    # plt.yticks(indices, [0, 1, 2])
    plt.xticks(indices, ['旅行', '运动', '聚餐','咖啡(厅)','人物','宠物','游乐园','办公室','汽车','车内'])
    plt.yticks(indices, ['旅行', '运动',