交叉熵的简单理解
交叉熵损失计算示例交叉熵损失公式交叉熵损失公式其中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 ·
254 阅读 ·
0 评论