Skip to content

Commit

Permalink
Update Docstring
Browse files Browse the repository at this point in the history
Co-authored-by: Kyle Daruwalla <daruwalla.k.public@icloud.com>
  • Loading branch information
shikhargoswami and darsnack authored Jan 30, 2021
1 parent a080ef3 commit eb3882d
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/losses/functions.jl
Original file line number Diff line number Diff line change
Expand Up @@ -432,7 +432,9 @@ end
Return the [focal_loss](https://arxiv.org/pdf/1708.02002.pdf)
Extremely useful for classification when you have highly imbalanced classes. It down-weights
well-classified examples and focuses on hard examples. Loss is much high for misclassified points as compared to well-classified points. Used in single-shot detectors.
well-classified examples and focuses on hard examples.
Loss is much high for misclassified points as compared to well-classified points.
Used in single-shot detectors.
"""
function focal_loss(ŷ, y; dims=1, agg=mean, 𝛄=2.0, eps = eps(eltype(ŷ)))
=.+ eps
Expand Down

0 comments on commit eb3882d

Please sign in to comment.