如何在Python中使用字符串轴而不是整数绘制混淆矩阵？

为了在Python中使用字符串轴而不是整数绘制混淆矩阵，可以按照以下步骤进行 –

• 为标签创建列表。
• 创建混淆矩阵。使用 confusion_matrix() 计算分类的准确性。
• 将 ~.axes.Axes 作为子图安排的一部分添加到图形中。
• 将2D矩阵或数组的值作为彩色图像绘制。
• 使用 colorbar() 方法，为 ScalarMappable 实例 mappable 创建一个色条。
• 使用 set_xticklabels 和 set_yticklabels 方法设置 x 和 y 轴的刻度标签。
• 使用 show() 方法显示图形。

阅读更多：Python 教程

示例

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()

输出