Fix ordinal mode to only be enabled when there are ordinal targets to…

… predict
creatis-myriad · Nov 24, 2023 · b71816d · b71816d
1 parent 21e29af
commit b71816d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/didactic/tasks/cardiac_multimodal_representation.py b/didactic/tasks/cardiac_multimodal_representation.py
@@ -205,6 +205,10 @@ def __init__(
                         accuracy, task="multiclass", num_classes=len(TABULAR_CAT_ATTR_LABELS[attr])
                     )
                 }
+        # Switch on ordinal mode if i) it's enabled, and ii) there are ordinal targets to predict
+        self.hparams.ordinal_mode = self.hparams.ordinal_mode and any(
+            attr in TabularAttribute.ordinal_attrs() for attr in self.predict_losses
+        )
 
         # Self-supervised losses and metrics
         self.contrastive_loss = None