利用python绘制混淆矩阵

利用python绘制混淆矩阵

利用python绘制混淆矩阵，直接附代码。

from sklearn.metrics import confusion_matrix
import pandas as pd
import matplotlib.pyplot as plt

y_pred = [] # ['2','2','3','1','4'] # 类似的格式
y_true = [] # ['0','1','2','3','4'] # 类似的格式
# 对上面进行赋值

C=confusion_matrix(y_true, y_pred, labels=['0','1','2','3','4']) # 可将'1'等替换成自己的类别，如'cat'。

plt.matshow(C, cmap=plt.cm.Reds) # Blues  Reds  Purples Greens
# plt.colorbar()

for i in range(len(C)):
    for j in range(len(C)):
        plt.annotate(C[j,i], xy=(i, j), horizontalalignment='center', verticalalignment='center')

plt.ylabel('True label')
plt.xlabel('Predicted label')
plt.show()

效果图：


如果需要切换不同的颜色，如：

参考：
python colorbar 颜色大全