如何在 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 次浏览

开启您的职业生涯

通过完成课程获得认证

立即开始
广告