We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 560e8ff commit 84535faCopy full SHA for 84535fa
segmentation_models_pytorch/losses/tversky.py
@@ -24,7 +24,7 @@ class TverskyLoss(DiceLoss):
24
ignore_index: Label that indicates ignored pixels (does not contribute to loss)
25
eps: Small epsilon for numerical stability
26
alpha: Weight constant that penalize model for FPs (False Positives)
27
- beta: Weight constant that penalize model for FNs (False Positives)
+ beta: Weight constant that penalize model for FNs (False Negatives)
28
gamma: Constant that squares the error function. Defaults to ``1.0``
29
30
Return:
0 commit comments