fudannlp16/focal-loss

二分类实现有bug

chenzk1993 opened this issue · 2 comments

chenzk1993 commented

二分类函数中 labels*(1-alpha)((1-y_pred)gamma) 是不是应该为 labels(1-alpha)((1-y_pred)**gamma)?

xbcReal commented

有道理

KevinChen1994 commented

我觉得不光是楼上提到的问题，我觉得应该是-labels * alpha * ((1 - y_pred) ** gamma) * tf.log(y_pred) - (1 - labels) * (1 - alpha) * (y_pred ** gamma) * tf.log(1 - y_pred).

  • 👍3