我正在尝试在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')
我是python的新手,所以我不确定自己在做什么错。
答案 0 :(得分:1)
在最后2行之前执行此操作:
class_names = class_names[::-1]
您只需要一个反向列表吗?