python 混淆矩阵记录

最新推荐文章于 2023-03-31 23:14:40 发布

小田同学tdx

最新推荐文章于 2023-03-31 23:14:40 发布

阅读量931

点赞数

分类专栏： Python

本文链接：https://blog.csdn.net/tiandd12/article/details/85136997

版权

Python 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

from sklearn.metrics import accuracy_score, confusion_matrix, f1_score
def plotCM(classes, matrix, savname):
    """classes: a list of class names"""
    # Normalize by row
    # matrix = matrix.astype(np.float)
    # linesum = matrix.sum(1)
    # linesum = np.dot(linesum.reshape(-1, 1), np.ones((1, matrix.shape[1])))
    # matrix /= linesum
    # plot
    plt.switch_backend('agg')
    fig = plt.figure()
    ax = fig.add_subplot(111)
    cax = ax.matshow(matrix)
    fig.colorbar(cax)
    ax.xaxis.set_major_locator(MultipleLocator(1))
    ax.yaxis.set_major_locator(MultipleLocator(1))
    for i in range(matrix.shape[0]):
        for j in range(matrix.shape[1]):
            print(matrix[i, j])
            ax.text(i, j, str('%.2f' % (matrix[i, j] * 100)), va='center', ha='center')
    ax.set_xticklabels([''] + classes, rotation=90)
    ax.set_yticklabels([''] + classes)
    #save

    # plt.imshow()
    plt.savefig(savname)
matrix = confusion_matrix(res['label'], res['pre_label'])
plotCM(['0','1'], matrix, 'res')