Merge pull request #132 from ImageMarkup/dice

Add support for Dice coefficient / F1 as a classification target metric
ImageMarkup · Feb 4, 2025 · e0ac502 · e0ac502
2 parents 1b9c3c4 + 3086100
commit e0ac502
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/isic_challenge_scoring/classification.py b/isic_challenge_scoring/classification.py
@@ -17,6 +17,7 @@ class ClassificationMetric(enum.Enum):
     BALANCED_ACCURACY = 'balanced_accuracy'
     AUC = 'auc'
     AVERAGE_PRECISION = 'ap'
+    DICE = 'dice'
 
 
 @dataclass(init=False)
@@ -100,6 +101,24 @@ def __init__(
                 ]
             )
             self.validation = per_category_auc.mean()
+        elif target_metric == ClassificationMetric.DICE:
+            self.overall = self.macro_average.at['dice']
+            per_category_dice = pd.Series(
+                [
+                    metrics.binary_dice(
+                        create_binary_confusion_matrix(
+                            truth_binary_values=truth_probabilities[category].gt(0.5).to_numpy(),
+                            prediction_binary_values=prediction_probabilities[category]
+                            .gt(0.5)
+                            .to_numpy(),
+                            weights=truth_weights['validation_weight'].to_numpy(),
+                            name=category,
+                        )
+                    )
+                    for category in categories
+                ]
+            )
+            self.validation = per_category_dice.mean()
 
     @staticmethod
     def _category_score(

diff --git a/tests/test_classification.py b/tests/test_classification.py
@@ -9,6 +9,7 @@
         ClassificationMetric.AUC,
         ClassificationMetric.BALANCED_ACCURACY,
         ClassificationMetric.AVERAGE_PRECISION,
+        ClassificationMetric.DICE,
     ],
 )
 def test_score(classification_truth_file_path, classification_prediction_file_path, target_metric):