将软标签转换为硬标签的 HardLabel 函数
HardLabel 函数的作用是将软标签转换为硬标签。软标签是一个二维列表,每一行表示一个样本的类别概率分布,每一列表示一个类别的概率。硬标签是一个一维列表,每个元素表示一个样本的类别。
函数首先获取软标签的样本数量和类别数量。然后,定义一个边界值,该值等于1除以类别数量。接下来,创建一个与样本数量相等的一维列表 hard_label,用于存储硬标签。
接下来,对于每个样本,获取其软标签。通过 torch.argmax 函数找到概率最大的类别作为预测类别,通过 torch.max 函数找到对应的概率值。如果预测的概率值大于边界值,则将预测类别作为硬标签的值;否则,将类别数量作为硬标签的值(表示未知类别)。
最后,返回硬标签列表。
原文地址: https://www.cveoy.top/t/topic/fcVI 著作权归作者所有。请勿转载和采集!