Python中的混淆矩阵

时间:2018-10-28 23:52:18

标签: python matplotlib scikit-learn

我正在尝试在python中创建一个混淆矩阵。但是,如您在下图中所看到的,我的班级标签arent彼此匹配。我本质上希望y轴以 Tshirt / Top 开头,以 Ankle Boots 结尾。这是我的代码:

test_labels = test_labels.argmax(axis=1)
predictions = predictions.argmax(axis=1)
cm = confusion_matrix(test_labels, predictions)

plt.setp(ax.xaxis.set_ticklabels(class_names), rotation=90, horizontalalignment='right')
plt.setp(ax.yaxis.set_ticklabels(class_names), rotation=360, horizontalalignment='right')

enter image description here

我是python的新手,所以我不确定自己在做什么错。

1 个答案:

答案 0 :(得分:1)

在最后2行之前执行此操作:

class_names = class_names[::-1]

您只需要一个反向列表吗?