如何在 Python 中绘制带有字符串轴而不是整数的混淆矩阵?
pythonmatplotlibserver side programmingprogramming
要在 Python 中绘制带有字符串轴而不是整数的混淆矩阵,我们可以采取以下步骤−
- 为标签制作列表。
- 创建混淆矩阵。使用 confusion_matrix() 计算分类准确率。
- 3. 将 '~.axes.Axes' 添加到图中作为子图排列的一部分。
- 将 2D 矩阵或数组的值绘制为彩色编码图像。
- 使用 colorbar() 方法,为 ScalarMappable 实例 *mappable* 创建颜色条>
- 6.使用 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()