Top-1 accuracy和Top-5 accuracy的概念及理解

LIsaWinLee發表於2020-10-31

官方解釋,也是我所查到了的最多的解釋

top-1

就是你預測的label取最後概率向量裡面最大的那一個作為預測結果,如果你的預測結果中概率最大的那個分類正確,則預測正確。否則預測錯誤

top-5

就是最後概率向量最大的前五名中,只要出現了正確概率即為預測正確。否則預測錯誤。

由此可以看出,top5一般比top1大

自行理解的解釋

在混淆矩陣中,可以看到每一個數字代表著每個真實類別被預測成所有類別的概率,那麼在這麼多預測結果中,按照概率由大到小排列。

top-1就是隻看概率最大的那個結果,是否預測正確,將其作為top-1的結果。

top-5就是看概率最大的前五個結果,只要裡面包含了預測正確的結果,就算預測正確,將其作為top-5的結果。

自我認為,自己這麼理解就明白了,兩種解釋語言表述方法,大家自行理解吧,若有更好的理解方式,歡迎交流~

相關文章