Pytorch NLLLOSS 的理解

Fri*_*eur 12 classification pytorch loss-function

PyTorch 的负对数似然损失nn.NLLLoss定义为:

\n

在此输入图像描述

\n

因此,如果使用单批次中一个的标准重量来计算损失,则损失的公式始终为:

\n
\n

-1 *(正确类别的模型预测)

\n
\n

例子:

\n

在此输入图像描述

\n

正确类别 = 0

\n

正确类别的模型预测 = 0.5

\n

损失 = -1 * 0.5

\n

那么,如果计算损失时没有涉及对数函数,为什么它被称为“负对数似然损失”呢?\n\xe2\x80\x8b

\n

Iva*_*van 17

事实上,没有log被用来计算结果,nn.NLLLoss所以这可能有点令人困惑。然而,我相信之所以这样称呼它是因为它期望接收对数概率:

通过前向调用给出的输入预计包含每个类别的对数概率。-文档

最后,将它包含在名称中没有多大意义,因为您可能还想将此函数应用于非对数概率......