resolve pb

aramis-lab · Sep 5, 2024 · 0479bf4 · 0479bf4
1 parent 44ca277
commit 0479bf4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/clinicadl/maps_manager/maps_manager.py b/clinicadl/maps_manager/maps_manager.py
@@ -467,6 +467,7 @@ def _check_args(self, parameters):
             )
 
         elif self.network_task == "regression" or self.network_task == "reconstruction":
+            self.n_classes = None
             self.metrics_module = MetricModule(
                 evaluation_metrics(self.network_task), n_classes=None
             )