[ENH] Add argument for axis labels to plot_scatter in _figures_for_st…

…atistic (#305) * add arg to plot * Automatic `pre-commit` fixes --------- Co-authored-by: TonyBagnall <[email protected]>
time-series-machine-learning · Nov 12, 2024 · 6448cb8 · 6448cb8
1 parent 63b53ea
commit 6448cb8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tsml_eval/evaluation/multiple_estimator_evaluation.py b/tsml_eval/evaluation/multiple_estimator_evaluation.py
@@ -1395,7 +1395,9 @@ def _figures_for_statistic(
                 f"{save_path}/{statistic_name}/figures/scatters/{est1}/", exist_ok=True
             )
 
-            scatter, _ = plot_pairwise_scatter(scores[:, i], scores[:, n], est1, est2)
+            scatter, _ = plot_pairwise_scatter(
+                scores[:, i], scores[:, n], est1, est2, metric=statistic_name.upper()
+            )
             scatter.savefig(
                 f"{save_path}/{statistic_name}/figures/scatters/{est1}/"
                 f"{eval_name}_{statistic_name.lower()}_scatter_{est1}_{est2}.pdf",