feat: Humanize plot names when converting to a CrossValidationItem

probabl-ai · Dec 20, 2024 · 46685f8 · 46685f8
1 parent c6fb23b
commit 46685f8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/examples/diagnose/plot_cross_validate.py b/examples/diagnose/plot_cross_validate.py
@@ -158,7 +158,7 @@
 # We can also access the plot after we have stored the ``CrossValidationReporter``:
 my_project.put("cross_validation_regression", reporter)
 cv_item = my_project.get_item("cross_validation_regression")
-cv_item.plots["compare_scores"]
+cv_item.plots["Scores"]
 
 # %%
 # .. note::

diff --git a/skore/src/skore/item/cross_validation_item.py b/skore/src/skore/item/cross_validation_item.py
@@ -153,8 +153,15 @@ def factory(cls, reporter: CrossValidationReporter) -> CrossValidationItem:
             "hash": _hash_numpy(X_array),
         }
 
+        humanized_plot_names = {
+            "compare_scores": "Scores",
+            "timing": "Timings",
+            "timing_normalized": "Normalized timings",
+        }
         plots_bytes = {
-            plot_name: plotly.io.to_json(plot, engine="json").encode("utf-8")
+            humanized_plot_names[plot_name]: (
+                plotly.io.to_json(plot, engine="json").encode("utf-8")
+            )
             for plot_name, plot in dataclasses.asdict(plots).items()
         }