我們在這里提供兩種不同的應(yīng)用場合：

1.如果你已經(jīng)通過實驗后有了矩陣數(shù)據(jù)，那么就可以直接利用這一方法，源碼如下：

from sklearn.metrics import confusion_matrix # 生成混淆矩陣函數(shù) import matplotlib.pyplot as plt # 繪圖庫 import numpy as npdef plot_confusion_matrix(cm, labels_name, title):cm = cm / cm.sum(axis=1)[:, np.newaxis] # 歸一化plt.imshow(cm, interpolation='nearest') # 在特定的窗口上顯示圖像plt.title(title) # 圖像標(biāo)題plt.colorbar()num_class = np.array(range(len(labels_name)))#獲取標(biāo)簽的間隔數(shù) plt.xticks(num_class, labels_name, rotation=90) # 將標(biāo)簽印在x軸坐標(biāo)上plt.yticks(num_class, labels_name) # 將標(biāo)簽印在y軸坐標(biāo)上plt.ylabel('True label') plt.xlabel('Predicted label')plt.show()#這里cm為你事先已經(jīng)獲得矩陣數(shù)據(jù)，一般為list類型 cm = [[5, 0, 0, 0],[1, 4, 0, 0],[0, 0, 5, 0],[1, 0, 0, 2]] cm = np.array(cm) #將list類型轉(zhuǎn)換成數(shù)組類型，如果已經(jīng)是numpy數(shù)組類型，則忽略此步驟。 labels_name = ['1','2','3','4']#這里個橫縱坐標(biāo)標(biāo)簽集合賦值 plot_confusion_matrix(cm,labels_name,"confusion_matrix")#調(diào)用函數(shù)

結(jié)果如圖：

2.如果你是在訓(xùn)練網(wǎng)絡(luò)的過程中想要在訓(xùn)練結(jié)束后繪制混淆矩陣，那么就需在上面的代碼上稍作修改：
?

from sklearn.metrics import confusion_matrix # 生成混淆矩陣函數(shù) import matplotlib.pyplot as plt # 繪圖庫 import numpy as np import tensorflow as tfdef plot_confusion_matrix(cm, labels_name, title):cm = cm / cm.sum(axis=1)[:, np.newaxis] # 歸一化plt.imshow(cm, interpolation='nearest') # 在特定的窗口上顯示圖像plt.title(title) # 圖像標(biāo)題plt.colorbar()num_class = np.array(range(len(labels_name)))#獲取標(biāo)簽的間隔數(shù) plt.xticks(num_class, labels_name, rotation=90) # 將標(biāo)簽印在x軸坐標(biāo)上plt.yticks(num_class, labels_name) # 將標(biāo)簽印在y軸坐標(biāo)上plt.ylabel('True label') plt.xlabel('Predicted label')plt.show()#這里通過list類型的標(biāo)簽數(shù)據(jù)來生成混淆矩陣 y_true = [.......] #這里要想辦法將你的實際標(biāo)簽類別轉(zhuǎn)換成list類型 pred_y = [.......] #這里要想辦法將你的預(yù)測標(biāo)簽類別轉(zhuǎn)換成list類型，一般網(wǎng)絡(luò)的最后一層通常是類別，這里根據(jù)你的網(wǎng)絡(luò)來獲得 cm = confusion_matrix(y_true,pred_y)cm = np.array(cm) #將list類型轉(zhuǎn)換成數(shù)組類型，如果已經(jīng)是numpy數(shù)組類型，則忽略此步驟。 labels_name = ['1','2','3','4']#這里個橫縱坐標(biāo)標(biāo)簽集合賦值 plot_confusion_matrix(cm,labels_name,"confusion_matrix")#調(diào)用函數(shù)

實驗結(jié)果與上圖類似。

總結(jié)

以上是生活随笔為你收集整理的利用python画混淆矩阵的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。