Refactor hierarchical measures calculation in MetricTemplate1.py

metric_template_1.py

@@ -102,18 +102,15 @@ class MetricTemplate1(evaluate.Metric):
             predictions
         )
 
-        ancestors = {  # The ancestors for each class, excluding the root
-            "G": {"B", "C", "E"},
-            "F": {"C"},
-        }
-
-        # Calculate hierarchical measures
-        hP, hR, hF = ham.hierarchical_precision_recall_fmeasure(
-            references, predictions, ancestors
+        # Example usage:
+        hierarchy = {"G": ["E"], "E": ["B"], "F": ["C"], "C": ["B"], "B": []}
+        # true_labels = [{'G'}]
+        # predicted_labels = [{'F'}]
+        hP, hR = ham.hierarchical_precision_recall(references, predictions, hierarchy)
+        hF = ham.hierarchical_f_measure(hP, hR)
+        print(
+            f"Hierarchical Precision: {hP}, Hierarchical Recall: {hR}, Hierarchical F-measure: {hF}"
         )
-        print(f"Hierarchical Precision (hP): {hP}")
-        print(f"Hierarchical Recall (hR): {hR}")
-        print(f"Hierarchical F-measure (hF): {hF}")
 
         return {
             "accuracy": accuracy,