混淆矩阵（confusion matrix）

混淆矩阵（Confusion Matrix）分析

https://blog.csdn.net/yutao03081/article/details/79111024

可视化混淆矩阵

import matplotlib.pyplot as plt
import numpy as np
from sklearn.metrics import confusion_matrix
def plot_confusion_matrix(confusion_mat):
    plt.figure(figsize=(50, 50))
    plt.imshow(confusion_mat)
    plt.title('Confusion Matrix')
    plt.colorbar()
    
#     labels = ['a', 'b', 'c', 'd']
    labels = list(range(confusion_mat.shape[0]))
    tick_marks = np.arange(len(labels))
    plt.xticks(tick_marks, labels)
    plt.yticks(tick_marks, labels)
    plt.ylabel('True Label')
    plt.xlabel('Predicted Label')
    plt.savefig('confusion_matrix.png')
    plt.show()
if __name__ == '__main__':
    y_true = [1, 0, 0, 2, 1, 0, 3, 3, 3]
    y_pred = [1, 1, 0, 2, 1, 0, 1, 3, 3]
#     confusion_mat = confusion_matrix(y_true, y_pred)
    confusion_mat = np.load('confusion_matrix.npy')
    print(confusion_mat.shape)
    plot_confusion_matrix(confusion_mat)