(1)直接输入混淆矩阵
import seaborn as sns
from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
def plot(matrix):
sns.set()
f,ax=plt.subplots()
print(matrix) #打印出来看看
sns.heatmap(matrix,annot=True,cmap="Blues",ax=ax) #画热力图
ax.set_title('confusion matrix') #标题
ax.set_xlabel('predict') #x轴
ax.set_ylabel('true') #y轴
matrix=np.array([[30 , 0 , 0 , 0 , 0],
[ 1 ,29 , 0 , 0 , 0],
[ 0 , 0 ,30 , 0 , 0],
[ 0 , 0 , 0 ,30 , 0],
[ 0 , 20 , 1, 1 ,28]])
plot(matrix)# 画原始的数据
matrix=matrix.astype('float') / matrix.sum(axis=1)[:, np.newaxis]
con_mat_norm = np.around(matrix, decimals=2)
plot(con_mat_norm)# 画比例数据