mob next [ci-skip] [ci skip] [skip ci]

lastFile:skore/src/skore/sklearn/_comparison/metrics_accessor.py
probabl-ai · Feb 6, 2025 · 4744202 · 4744202
1 parent fab3a0b
commit 4744202
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/skore/src/skore/sklearn/_comparison/metrics_accessor.py b/skore/src/skore/sklearn/_comparison/metrics_accessor.py
@@ -162,12 +162,15 @@ def _compute_metric_scores(
                 results.append(result)
                 progress.update(main_task, advance=1, refresh=True)
 
-            results = pd.concat(
-                results,
-                axis=0,
-                keys=[f"#{i}" for i in range(len(self._parent.estimator_reports_))],
+            results = pd.concat(results, axis=0, ignore_index=True)
+            results.index = pd.MultiIndex.from_tuples(
+                (
+                    (i, report.estimator_name_)
+                    for i, report in enumerate(self._parent.estimator_reports_)
+                ),
+                names=[None, "Estimator"],
             )
-            results = results.swaplevel(0, 1)
+
             if aggregate:
                 if isinstance(aggregate, str):
                     aggregate = [aggregate]