对火炬NLLLOS的理解 [英] Understanding of Pytorch NLLLOSS
本文介绍了对火炬NLLLOS的理解的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
PyTorch的负对数似然损失nn.NLLLoss
定义为:
因此,如果损失是以一批一的标准重量计算的,则损失的公式始终为:
-1*(模型对正确类的预测)
示例:
正确类=0
正确类别的模型预测=0.5
损耗=-1*0.5
那么,如果不存在计算损失的对数函数,为什么它被称为负对数似然损失? 推荐答案
实际上没有使用log
来计算nn.NLLLoss
的结果,因此这可能会有点令人困惑。不过,我相信之所以这样调用,是因为它期望接收对数概率:
通过前向呼叫提供的输入预计将包含每个类别的对数概率。-docs
归根结底,将其包含在名称中没有多大意义,因为您可能希望将此函数应用于非对数概率...
这篇关于对火炬NLLLOS的理解的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文