关于topk accuracy

这个概念是单标签多分类中的,先说我们认识的acc,这个是求softmax后的最大值索引,只要对应上就是预测正确。

这就是top1,topk啥意思呢?

最大值的索引可能与标签没有对应上,只要最大值索引排序中前k个有对应的正确标签就说明预测对了,举个例子:

softmax后y_scores为:

[0.2,0.3,0.4,0.1]

[0.15,0.3,0.05,0.5]

求argmax后为

[2]

[3]

正确标签y为

[0,1,0,0]

[0,1,0,0]

argmax后

[1]

[1]

top1直接为0,因为最大值索引对应不上。

但top2为100%,最大值排序后相应的索引顺序

[2,1,0,3]

[3,1,0,2]

索引都是第二个正确,那top2 acc肯定是100%了。

另外有相关问题可以加入QQ群讨论,不设微信群

QQ群:868373192 

语音图像视频深度-学习群

发布了252 篇原创文章 · 获赞 246 · 访问量 9万+

猜你喜欢

转载自blog.csdn.net/SPESEG/article/details/105354352