如何讀取張量流混淆矩陣行和列

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的未分類元素相對應，依此類推。