pytorch损失函数:MSE,NLLLose,Cross Entropy,KLDivLoss

使用说明

如果想处理分类任务,请看1。
如果想处理向量间的距离,请看2。

1. NLLLose与Cross Entropy

他们的输入是(概率,标签)这样的二元组。
一个例子:假设三分类任务最后的输出是这三个数 [0.8, 0.2, 0.7]。标签是0。
想要用cross entropy计算损失,那么我们直接把这两个丢进去就好了,不用做任何处理,loss = cross_entropy( probability, label )。
想要丢到NLLLose函数中计算损失的话,我们首先要对 [0.8, 0.2, 0.7] 计算softmax,假设结果是 [0.45, 0, 0.40, 0.15], 然后再求个对他的每个概率求个log,然后把处理后的概率丢到函数里,计算 loss = NLLLose( probability, label )。

2. MSE与KLDivLoss

他们的输入是一串向量,label也是一串向量,所以可以用来衡量两串向量间的接近程度。

经过数学证明,KLDiv是恒大于0的,可以放心使用。至于它和MSE之间的效果区别,笔者目前未知,大家可留言分享。

KLDiv还会用在label smoothing中。

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值