From 9ab4f10269f269ba10f08cfef2763649b175e3b4 Mon Sep 17 00:00:00 2001
From: jpaillard <joseph.paillard@inria.fr>
Date: Fri, 25 Oct 2024 14:41:12 +0200
Subject: [PATCH] update figure legend

---
 examples/plot_variable_importance_classif.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/examples/plot_variable_importance_classif.py b/examples/plot_variable_importance_classif.py
index d65006f..f158813 100644
--- 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"]: