fudannlp16 / focal-loss

Tensorflow version implementation of focal loss for binary and multi classification

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

typo in focal_loss?

ddeeppnneett opened this issue · comments

Line 25 in focal_loss.py,

(1-y_pred)*gamma
or
(1-y_pred)**gamma