some changes

clinicadl/utils/metric_module.py

@@ -480,6 +480,12 @@ def compute_ap(y, y_pred, *args):
 
         return average_precision_score(y, y_pred, *args)
 
+    @staticmethod
+    def compute_roc_auc(y_pred, y, *args):
+        from monai.metrics.rocauc import compute_roc_auc
+
+        return compute_roc_auc(y_pred, y, *args)
+
 
 class RetainBest:
     """

clinicadl/utils/task_manager/classification.py

@@ -53,6 +53,7 @@ def evaluation_metrics(self):
             "MK",
             "LR_plus",
             "LR_minus",
+            "roc_auc",
         ]
 
     @property

clinicadl/utils/task_manager/task_manager.py

@@ -18,9 +18,11 @@
 # TODO: add function to check that the output size of the network corresponds to what is expected to
 #  perform the task
 class TaskManager:
-    def __init__(self, mode: str, n_classes: int = None):
+    def __init__(self, mode: str, n_classes: int = 2):
         self.mode = mode
-        self.metrics_module = MetricModule(self.evaluation_metrics, n_classes=n_classes)
+        self.metrics_module = MetricModule(
+            v=self.evaluation_metrics, n_classes=n_classes
+        )
 
     @property
     @abstractmethod