commenting out auroc

mlcommons · Apr 18, 2024 · 80e1aaf · 80e1aaf
1 parent 4d751fd
commit 80e1aaf
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/GANDLF/metrics/classification.py b/GANDLF/metrics/classification.py
@@ -62,11 +62,11 @@ def overall_stats(prediction: torch.Tensor, target: torch.Tensor, params: dict)
                 num_classes=params["model"]["num_classes"],
                 average=average_type_key,
             ),
-            "aucroc": tm.AUROC(
-                task=task,
-                num_classes=params["model"]["num_classes"],
-                average=average_type_key if average_type_key != "micro" else "macro",
-            ),
+            # "aucroc": tm.AUROC(
+            #     task=task,
+            #     num_classes=params["model"]["num_classes"],
+            #     average=average_type_key if average_type_key != "micro" else "macro",
+            # ),
         }
         for metric_name, calculator in calculators.items():
             if metric_name == "aucroc":