Update multiclass_specificity_macro.py

multiclass_specificity_macro.py

@@ -87,25 +87,8 @@ class multiclass_specificity_macro(evaluate.Metric):
         pass
 
     def _compute(self, predictions, references):
-        import numpy as np
-        """Returns the scores"""
-        # TODO: Compute the different scores of the module
-        unique_classes = np.unique(predictions)
-        num_classes = len(unique_classes)
-        specificity = np.zeros(num_classes)
-        class_counts = np.bincount(predictions)
-        total_samples = len(predictions)
-
-        for i, class_label in enumerate(unique_classes):
-            true_negative = sum((predictions != class_label) & (references != class_label))
-            total_negative = sum(predictions != class_label)
-            
-            if total_negative != 0:
-                specificity[i] = true_negative / total_negative
-            else:
-                specificity[i] = 0.0
-        macro_specificity = np.mean(specificity)
+        from imblearn.metrics import specificity_score
     
         return {
-            "macro_specificity": macro_specificity,
+            "macro_specificity": specificity_score(references, predictions, average = "macro")
         }