From fc921d4d6ba4bace6a02c8ab563a104564a8fce9 Mon Sep 17 00:00:00 2001 From: "Viet-Anh, Nguyen" Date: Thu, 21 Apr 2022 14:40:48 +0700 Subject: [PATCH] Correct FNs -> False Negatives --- segmentation_models_pytorch/losses/tversky.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/segmentation_models_pytorch/losses/tversky.py b/segmentation_models_pytorch/losses/tversky.py index 7d6e1038..4850af6d 100644 --- a/segmentation_models_pytorch/losses/tversky.py +++ b/segmentation_models_pytorch/losses/tversky.py @@ -24,7 +24,7 @@ class TverskyLoss(DiceLoss): ignore_index: Label that indicates ignored pixels (does not contribute to loss) eps: Small epsilon for numerical stability alpha: Weight constant that penalize model for FPs (False Positives) - beta: Weight constant that penalize model for FNs (False Positives) + beta: Weight constant that penalize model for FNs (False Negatives) gamma: Constant that squares the error function. Defaults to ``1.0`` Return: