wzzzd / text_classifier_pytorch

基于Pytorch的文本分类框架,支持TextCNN、Bert、Electra等。
57 stars 9 forks source link

focal loss的权重感觉没调啊 #5

Open zgctmac opened 1 year ago

zgctmac commented 1 year ago

看了下代码,感觉focal_loss这一部分的权重并没有调

wzzzd commented 1 year ago

展开说说看

zgctmac commented 1 year ago

focal_loss.py里面没看见对alpha的调参

wzzzd commented 1 year ago

确实没有,考虑到alpha如果作为一个超参来配置,那么这样的话代码兼容性可能不太好处理(比如alpha值需要根据具体场景来配置,且二分类和多分类的情形下,alpha的长度也不同)。

这边可以人为修改focal_loss.py中weight的默认值,来达到想要的效果。