FocalLoss#
- class torch_uncertainty.losses.FocalLoss(gamma, alpha=None, reduction='mean')[source]#
Focal-Loss for classification tasks.
- Parameters:
gamma (float, optional) – A constant, as described in the paper.
alpha (Tensor, optional) – Weights for each class. Defaults to
None
.reduction (str, optional) –
'mean'
,'sum'
or'none'
. Defaults to'mean'
.
References
[2] Inspired by AdeelH/pytorch-multi-class-focal-loss .