【pytorch损失函数之nn.CrossEntropyLoss()、nn.NLLLoss()】
文章目录一、前言二、理论三、项目
一、前言
nn.CrossEntropyLoss()这个损失函数⽤于多分类问题虽然说的是交叉熵。
nn.CrossEntropyLoss()是nn.logSoftmax()和nn.NLLLoss()的整合,可以直接使⽤它来替换⽹络中的这两个操作。
二、理论
⾸先输⼊是size是(minibatch,C)。这⾥的C是类别数。损失函数的计算如下:
损失函数中也有权重weight参数设置 若设置权重,则公式为:
其他参数不具体说,和设置差不多,默认情况下,对minibat
原创
2022-05-03 17:11:07 ·
1594 阅读 ·
0 评论