MSE Loss、BCE Loss

部署运行你感兴趣的模型镜像

MSE 均方误差

pytorch官网中的介绍
在这里插入图片描述
measures the mean squared error (squared L2 norm) between each element in the input x x x and target y y y.
即均方误差,回归问题解决的是对具体数值的预测,使用于回归问题下的损失函数。
使用示例:

loss = nn.MSELoss()
input = torch.randn(3, 5, requires_grad=True)
target = torch.randn(3, 5)
output = loss(input, target)
output.backward()

使用MSE损失函数,受离群点的影响较大。

MAE 平均绝对误差

MAE受离群点的影响较小,拟合直线能较好的表征正常数据的分布情况。

从求解梯度的复杂度来讲,MSE要优于MAE,能够较快达到收敛。

BCE

CE主要适用于二分类的任务,而且多标签分类任务可以简单地理解为多个二元分类任务叠加。所以BCE经过简单修改也可以适用于多标签分类任务。使用BCE之前,需要将输出变量量化在[0,1]之间(可以使用Sigmoid激活函数)

参考

【损失函数】MSE, MAE, Huber loss详解

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

所谓远行Misnearch

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值