Skip to content

Commit f58fd6d

Browse files
authored
Update the description of TverskyLoss (#626)
1 parent b5981e7 commit f58fd6d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

segmentation_models_pytorch/losses/tversky.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
class TverskyLoss(DiceLoss):
1212
"""Tversky loss for image segmentation task.
13-
Where TP and FP is weighted by alpha and beta params.
13+
Where FP and FN is weighted by alpha and beta params.
1414
With alpha == beta == 0.5, this loss becomes equal DiceLoss.
1515
It supports binary, multiclass and multilabel cases
1616

0 commit comments

Comments
 (0)