Add helper function for labels with isco and title

isco_hierarchical_accuracy.py

@@ -264,9 +264,16 @@ class ISCO_Hierarchical_Accuracy(evaluate.Metric):
         print("Weighted ISCO hierarchy dictionary created as isco_hierarchy")
         # print(self.isco_hierarchy)
 
+    # Define the mapping from ISCO_CODE_TITLE to ISCO codes
+    def _extract_isco_code(isco_code_title: str):
+        # ISCO_CODE_TITLE is a string like "7412 Electrical Mechanics and Fitters" so we need to extract the first part for the evaluation.
+        return isco_code_title.split()[0]
+
     def _compute(self, predictions, references):
         """Returns the accuracy scores."""
         # Convert the inputs to strings
+        if len(predictions[0]) > 4:
+            predictions = [self._extract_isco_code(p) for p in predictions]
         predictions = [str(p) for p in predictions]
         references = [str(r) for r in references]