PredictFilter 函数使用训练好的分类模型对开放特征进行预测,并根据给定的阈值'theta'过滤掉置信度低于阈值的预测结果。具体步骤如下:

  1. 将分类模型和开放特征移动到指定的设备(dev)上。
  2. 使用分类模型对开放特征进行预测,得到待分配的标签(wait_to_dis_label)。
  3. 将待分配的标签转换为概率形式,使用Logits2Soft函数。
  4. 如果给定的阈值'theta'小于0,则找到最大概率值和对应的预测标签,将'theta'设置为最大概率值的中位数。
  5. 遍历每个预测结果,如果预测概率低于阈值'theta',则将该结果替换为平均概率张量(average_tensor)的克隆。
  6. 返回经过过滤的预测结果(wait_to_dis_label)。

需要注意的是,这里的'average_tensor'是一个长度为'class_cat'的张量,每个元素的值都为1.0/'class_cat',表示对每个类别的平均概率。这样做的目的是在过滤掉置信度低的预测结果后,将这些结果平均分配给所有类别,以保持概率分布的一致性。

PredictFilter 函数:基于阈值的预测结果过滤

原文地址: http://www.cveoy.top/t/topic/fdep 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录