fastai 混淆矩阵类不按顺序排列

Question

我正在使用 fastai 绘制我的 fastai 模型的混淆矩阵。我正在使用此代码：

interp = ClassificationInterpretation.from_learner(learn)
interp.plot_confusion_matrix()

我的班级是[0,1,2,3,4,5,6,7,8,9,10,11,12,13]。

混淆矩阵无序打印类，例如：[0,1,10,11,12,13,4,5,6,7,8,9]

我该如何解决这个问题？

Answer 1

一个简单的替代方法是通过为类添加前导零（而不是使用数字）来更改您的类别：

• 1 => 01
• 2 => 02
• ...
• 10 => 10

您可以像这样使用02d 格式：

"{:02d}".format(category) 甚至更好（Python 3.x）：`f"{category:02d}"。

这样，您可以确定02 在10 之前。