Pythonで整数ではなく文字列軸を使用して混同行列をプロットするにはどうすればよいですか？

Pythonで整数ではなく文字列軸を使用して混同行列をプロットするには、次の手順を実行できます-

• ラベルのリストを作成します。
• 混同行列を作成します。 confusion_matrix（）を使用します 分類の精度を計算します。
• 3。 '〜.axes.Axes'を追加します サブプロットの配置の一部として図に。
• 2D行列または配列の値を色分けされた画像としてプロットします。
• colorbar（）の使用 メソッド、 ScalarMappableのカラーバーを作成します インスタンス、*マップ可能*
• 6。 set_xticklabelsを使用してxおよびyティックラベルを設定します およびset_yticklabelsメソッド。
• 図を表示するには、 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()

出力