如何在 Python 中绘制带有字符串轴而不是整数的混淆矩阵?

pythonmatplotlibserver side programmingprogramming

要在 Python 中绘制带有字符串轴而不是整数的混淆矩阵,我们可以采取以下步骤−

  • 为标签制作列表。
  • 创建混淆矩阵。使用 confusion_matrix() 计算分类准确率。
  • 3. 将 '~.axes.Axes' 添加到图中作为子图排列的一部分。
  • 将 2D 矩阵或数组的值绘制为彩色编码图像。
  • 使用 colorbar() 方法,为 ScalarMappable 实例 *mappable* 创建颜色条>
  • 6.使用 set_xticklabelsset_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()

输出


相关文章