update figure legend

mind-inria · Oct 25, 2024 · 9ab4f10 · 9ab4f10
1 parent 7bf4d86
commit 9ab4f10
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/plot_variable_importance_classif.py b/examples/plot_variable_importance_classif.py
@@ -235,7 +235,7 @@
     importance_arr[:, 0, :],
     positions=np.arange(1, n_features + 1) - 0.25,
     widths=0.2,
-    label="Linear",
+    label="CPI Linear",
     vert=False,
 )
 for item in ["whiskers", "fliers", "medians", "caps", "boxes"]:
@@ -245,7 +245,7 @@
     importance_arr[:, 1, :],
     positions=np.arange(1, n_features + 1),
     widths=0.2,
-    label="Poly",
+    label="CPI Poly",
     vert=False,
 )
 for item in ["whiskers", "fliers", "medians", "caps", "boxes"]:
@@ -255,7 +255,7 @@
     importance_arr[:, 2, :],
     positions=np.arange(1, n_features + 1) + 0.25,
     widths=0.2,
-    label="Permutation",
+    label="PI Poly",
     vert=False,
 )
 for item in ["whiskers", "fliers", "medians", "caps", "boxes"]: