交叉熵的简单理解
交叉熵损失计算示例交叉熵损失公式
交叉熵损失公式
其中y为label,p^为预测的正类别概率,即在二分类中通过sigmoid函数得出的正类别概率大小。
举例:
criterion = nn.CrossEntropyLoss()
input = torch.randn(3, 5, requires_grad=True)
target = torch.empty(3, dtype=torch.long).random_(5)
loss = criterion(input, target)
上述代码即为求
原创
2021-09-24 19:25:02 ·
225 阅读 ·
0 评论