Open zgctmac opened 1 year ago
看了下代码,感觉focal_loss这一部分的权重并没有调
展开说说看
focal_loss.py里面没看见对alpha的调参
确实没有,考虑到alpha如果作为一个超参来配置,那么这样的话代码兼容性可能不太好处理(比如alpha值需要根据具体场景来配置,且二分类和多分类的情形下,alpha的长度也不同)。
这边可以人为修改focal_loss.py中weight的默认值,来达到想要的效果。
看了下代码,感觉focal_loss这一部分的权重并没有调