search

fudannlp16 / focal-loss

Tensorflow version implementation of focal loss for binary and multi classification
108 stars 27 forks source link

typo in focal_loss? #3

Open ddeeppnneett opened 6 years ago

ddeeppnneett commented 6 years ago

Line 25 in focal_loss.py,

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