pytorch中CrossEntropyLoss中weight的问题

pytorch中CrossEntropyLoss中weight的问题

由于研究的需要,最近在做一个分类器,但类别数量相差很大。
在查看nn.CrossEntropyLoss()的官方文档时看到这么一句
在这里插入图片描述
感觉有救了,遂想应用到我自己的网络中,但是weight是自己初始化的。怎么初始化又有问题了。在看focal loss的时候有点印象,样本多的类别权重应该小,但是有没有一个公式啊?
在中文互联网是找了很久,没有太多的帖子,相关问题的帖子文章很少,于是就去谷歌了。
在以下两篇帖子找到了答案。
先放链接
https://discuss.pytorch.org/t/weights-in-weighted-loss-nn-crossentropyloss/69514
帖子1
https://datascience.stackexchange.com/questions/48369/what-loss-function-to-use-for-imbalanced-classes-using-pytorch
帖子2
第一篇帖子中有两个公式

  • 8
    点赞
  • 28
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值