如何在 Python 中使用字串軸而不是整數繪製混淆矩陣？

要在 Python 中使用字串軸而不是整數繪製混淆矩陣，我們可以採取以下步驟-

  • 建立一個標籤列表。
  • 建立一個混淆矩陣。使用 confusion_matrix() 計算分類的準確性。
  • 把 '~.axes.Axes' 作為子圖排列的一部分新增到圖形中。
  • 用顏色編碼影像繪製二維矩陣或陣列的值。
  • 使用 colorbar() 方法為 ScalarMappable 例項 *mappable* 建立一個顏色條。
  • 使用 set_xticklabels 和 set_yticklabels 方法設定 x 和 y 刻度標籤。
  • 要顯示圖形，請使用 show() 方法。

示例程式碼

from matplotlib import pyplot as plt
from sklearn.metrics import confusion_matrix
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
labels = ['business', 'health']
cm = confusion_matrix([3], [2])
fig = plt.figure()
ax = fig.add_subplot(111)
cax = ax.matshow(cm)
fig.colorbar(cax)
ax.set_xticklabels([''] + labels)
ax.set_yticklabels([''] + labels)
plt.show()

輸出

更新於： 2021 年 5 月 6 日

558 次瀏覽

開啟您的職業生涯

透過完成課程獲得認證

立即開始
廣告