search

wzzzd / text_classifier_pytorch

基于Pytorch的文本分类框架，支持TextCNN、Bert、Electra等。
58 stars 10 forks source link

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

Open zgctmac opened 2 years ago

zgctmac commented 2 years ago

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

wzzzd commented 2 years ago

展开说说看

zgctmac commented 2 years ago

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

wzzzd commented 2 years ago

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

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