对火炬NLLLOS的理解 [英] Understanding of Pytorch NLLLOSS

问题描述

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

因此，如果损失是以一批一的标准重量计算的，则损失的公式始终为：

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

示例：

正确类=0

正确类别的模型预测=0.5

损耗=-1*0.5

那么，如果不存在计算损失的对数函数，为什么它被称为负对数似然损失？ ​

推荐答案

实际上没有使用log来计算nn.NLLLoss的结果，因此这可能会有点令人困惑。不过，我相信之所以这样调用，是因为它期望接收对数概率：

通过前向呼叫提供的输入预计将包含每个类别的对数概率。-docs

归根结底，将其包含在名称中没有多大意义，因为您可能希望将此函数应用于非对数概率...

这篇关于对火炬NLLLOS的理解的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！