我使用ResNet9算法构建了这个机器学习模型。该模型有38个类别。预测后,它会返回一个如下所示的元组:
它返回的是负值。在上面的例子中,最高值为-204.0966,这意味着它最有可能被预测为正确类别。
我想将其转换为百分比,以便查看该图像的置信度(准确性)。如果有人想查看代码,请告诉我。
回答:
你应该应用Softmax函数,该函数将向量值映射到总和为1的值(这是一种获得概率分布的方法),然后将结果乘以100以得到百分比:
import torchtorch.nn.functional.softmax(model_output) * 100