add macro scores

matching_series.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 """TODO: Add a description here."""
 
+import statistics
+
 import datasets
 import evaluate
 import numpy as np
@@ -153,7 +155,9 @@ class matching_series(evaluate.Metric):
             covered_mse_features.append(covered_mse_f)
             harmonic_mean_features.append(harmonic_mean_f)
             index_mse_features.append(index_mse_f)
-
+        macro_matching_mse = statistics.mean(matching_mse_features)
+        macro_covered_mse = statistics.mean(covered_mse_features)
+        macro_harmonic_mean = statistics.mean(harmonic_mean_features)
         return {
             "matching_mse": matching_mse,
             "harmonic_mean": harmonic_mean,
@@ -163,4 +167,7 @@ class matching_series(evaluate.Metric):
             "harmonic_mean_features": harmonic_mean_features,
             "covered_mse_features": covered_mse_features,
             "index_mse_features": index_mse_features,
+            "macro_matching_mse": macro_matching_mse,
+            "macro_covered_mse": macro_covered_mse,
+            "macro_harmonic_mean": macro_harmonic_mean,
         }