如何读取张量流混淆矩阵行和列

Question

对于我的2个类（ 1 = [0, 1]和0 = [1, 0] ），我们使用tf.confusion_matrix查找该模型的混淆矩阵。 我的结果之一类似于下面的验证集：

[ [1800  17] 
  [283  600] ]

搜索后，我看到的阅读类型不止一种，其中有些人说[[TN FP][FN TP]] ，但另一些人则以这种方式阅读[[TP FP][FN TN]] ，我很困惑哪一个适合我的情况？ 如果可以的话，请给我一个取决于科学研究的答案。

Answer 1

类标签应从0开始。例如，如果num_classes为3，则可能的标签为[0, 1, 2] 。 请注意，假定可能的标签为[0, 1, 2, 3, 4] ，从而导致5x5混淆矩阵。

因此最好不要将一个热张量传递给该函数;）（tf.argmax在这里可能是个好朋友）

这意味着第一个元素（行0 col 0）对应于已为类别0正确分类的元素数量。

第0行第1行将与类别0的未分类元素相对应，依此类推。