学习笔记30-Top1和Top5定义与代码复现

定义

Top-1： Accuracy是指排名第一的类别与实际结果相符的准确率，就是你预测的label取最后概率向量里面最大的那一个作为预测结果，如过你的预测结果中概率最大的那个分类正确，则预测正确。否则预测错误。
Top-5： Accuracy是指排名前五的类别包含实际结果的准确率，就是最后概率向量最大的前五名中，只要出现了正确概率即为预测正确。否则预测错误。
TOP-5正确率=（所有测试图片中正确标签包含在前五个分类概率中的个数）除以（总的测试图片数）
TOP-5错误率=（所有测试图片中正确标签不在前五个概率中的个数）除以（总的测试图片数）
注意： 我们平时说的top1就是准确率，Accuracy和F1-Score这些是判断分类模型总体的标准。

代码复现

输入是模型输出（batch_size×num_of_class），目标label（num_of_class向量），元组（分别向求top几）

acc.py

import torch
def accu(output, target, topk=