混淆矩阵demo

import seaborn as sns
import pandas as pd
from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
C = [[0 for col in range(3)] for row in range(3)]

sns.set()
f,ax=plt.subplots()
y_true = [0,0,1,2,1,2,0,2,2,0,1,1]
y_pred = [1,0,1,2,1,0,0,2,2,0,1,1]
#C2 = confusion_matrix(y_true, y_pred, labels=[0, 1, 2])
C2 = confusion_matrix(y_true, y_pred)
C2 = C2.astype('float') / C2.sum(axis=1)[:, np.newaxis]
C = C + C2
print(C2) #打印出来看看
sns.heatmap(C,annot=True,ax=ax) #画热力图

ax.set_xticklabels(['a','b','c'])
ax.set_yticklabels(['a','b','c'])

ax.set_title('confusion matrix') #标题
ax.set_xlabel('predict') #x轴
ax.set_ylabel('true') #y轴


plt.show()