nn.BCEWithLogitsLoss中weight参数和pos_weight参数的作用及用法
如果我们在做多分类任务,有些类比较重要,有些类不太重要,想要模型更加关注重要的类别,那么只需将比较重要的类所对应的w权重设置大一点,不太重要的类所对应的w权重设置小一点。其中pos_weight对应上式公式中的p_c,这个参数是为了调节正负样本不均衡问题的,如果正负样本比是10:1,那么我们可以将p_c设置为1/10来平衡正负样本。上式是nn.BCEWithLogitsLoss损失函数的计算公式,其中w_n对应weight参数。
原创
2024-01-12 13:56:51 ·
1628 阅读 ·
0 评论