Pytorch 加权BCE损失

最新推荐文章于 2024-07-27 17:29:13 发布

镇长1998

最新推荐文章于 2024-07-27 17:29:13 发布

阅读量2.6k

点赞数 5

分类专栏：实验室文章标签： PyTorch BCELoss 权重调整二分类损失函数

本文链接：https://blog.csdn.net/weixin_41514525/article/details/115495450

版权

实验室专栏收录该内容

41 篇文章 1 订阅

订阅专栏

bce_loss = nn.BCELoss(reduction='none')  # 默认选项是mean, 设置为none后会返回一个和target一样尺寸的tensor, 每个位置的数字对应网络输出和target对应位置上的loss

weight = torch.zeros_like(target)
weight = torch.fill_(weight, 0.2)
weight[target > 0] = 0.8

bce_loss = bce_loss(input.float(), target.float())
bce_loss = torch.sum(bce_loss * weight)    # torch.sum() 或者 torch.mean() 都可以看个人选择