Merge nucleic/sleek-ember-seal-uady into dev

2026-07-31 14:30:10 -07:00
parent 4ed4763557
commit 3d35f4953f
5 changed files with 248 additions and 15 deletions
@@ -274,6 +274,20 @@ def multitask_metrics(
            targets.secondary[targets.secondary_mask].tolist(),
            secondary_predictions[targets.secondary_mask].tolist(),
        )
+        supported_labels = [
+            label
+            for label, row in zip(
+                secondary["confusionMatrix"]["labels"],
+                secondary["confusionMatrix"]["rows"],
+            )
+            if sum(row)
+        ]
+        secondary["supportedLabels"] = supported_labels
+        secondary["supportedMacroRecall"] = float(
+            np.mean(
+                [secondary["perPurposeRecall"][label] for label in supported_labels]
+            )
+        )
    else:
        secondary = None