cd dias

time-series-machine-learning · Nov 17, 2023 · 92e75ac · 92e75ac
1 parent 6d510c3
commit 92e75ac
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/tsml_eval/evaluation/multiple_estimator_evaluation.py b/tsml_eval/evaluation/multiple_estimator_evaluation.py
@@ -1,6 +1,7 @@
 """Functions for evaluating multiple estimators on multiple datasets."""
 
 import os
+import pickle
 from datetime import datetime
 
 import numpy as np
@@ -793,7 +794,17 @@ def _figures_for_statistic(
 
     cd = plot_critical_difference(scores, estimators, errors=not higher_better)
     cd.savefig(
-        f"{save_path}/{statistic_name}/figures/{statistic_name}_critical_difference.png"
+        f"{save_path}/{statistic_name}/figures/"
+        f"{statistic_name}_critical_difference.png",
+        bbox_inches="tight",
+    )
+    pickle.dump(
+        cd,
+        open(
+            f"{save_path}/{statistic_name}/figures/"
+            f"{statistic_name}_critical_difference.pickle",
+            "wb",
+        ),
     )