【Python机器学习】实现二分类混淆矩阵可视化

Rssevenn

已于 2022-03-24 20:12:45 修改

阅读量3.1k

点赞数 4

分类专栏： Python机器学习文章标签： python 机器学习分类 sklearn

于 2022-03-09 11:57:13 首次发布

本文链接：https://blog.csdn.net/m0_61729283/article/details/123374017

版权

Python机器学习专栏收录该内容

1 篇文章 0 订阅

订阅专栏

直接上代码

def cm_plot(y, yp):
    '''
    y: 真实值
    y_pred:预测值
    '''
    '''混淆矩阵绘画'''
    from sklearn.metrics import confusion_matrix  # 导入混淆矩阵函数

    cm = confusion_matrix(y, yp)  # 混淆矩阵
    print('混淆矩阵为：\n',cm)
    import matplotlib.pyplot as plt  # 导入作图库
    plt.matshow(cm, cmap=plt.cm.Greens)  # 画混淆矩阵图，配色风格使用cm.Greens，更多风格请参考官网。
    plt.colorbar()  # 颜色标签

    for x in range(len(cm)):  # 数据标签
        for y in range(len(cm)):
            plt.annotate(cm[x, y], xy=(y, x),verticalalignment='center',horizontalalignment='center')
            # 这边一个重要的易错点就是关于xy需要添加数值的位置的点要换成（y, x），因为矩阵可视化的xy与实际意义上的坐标是相反的
    plt.ylabel('True label')  # 坐标轴标签
    plt.xlabel('Predicted label')  # 坐标轴标签
    plt.show()