【发布时间】:2021-11-18 08:56:24
【问题描述】:
PyTorch 的负对数似然损失,nn.NLLLoss 定义为:
因此,如果使用单批次中一个的标准重量计算损失,则损失的公式始终为:
-1 *(正确类别的模型预测)
示例:
正确类别 = 0
模型预测正确 class= 0.5
损失 = -1 * 0.5
那么,如果在计算损失时不涉及对数函数,为什么将其称为“负对数似然损失”?
【问题讨论】:
标签: pytorch classification loss-function