From 8286f53ae788c045446200901012ca0a89e96f57 Mon Sep 17 00:00:00 2001 From: jpaillard Date: Mon, 17 Feb 2025 19:15:14 +0100 Subject: [PATCH] remove functional implem in example --- examples/plot_diabetes_variable_importance_example.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/examples/plot_diabetes_variable_importance_example.py b/examples/plot_diabetes_variable_importance_example.py index 8039221..46835d0 100644 --- a/examples/plot_diabetes_variable_importance_example.py +++ b/examples/plot_diabetes_variable_importance_example.py @@ -55,7 +55,7 @@ from sklearn.metrics import r2_score, root_mean_squared_error from sklearn.model_selection import KFold -from hidimstat import CPI, LOCO, permutation_importance +from hidimstat import CPI, LOCO, PermutationImportance ############################################################################# # Load the diabetes dataset @@ -157,15 +157,14 @@ print(f"Fold {i}") X_train, X_test = X[train_index], X[test_index] y_train, y_test = y[train_index], y[test_index] - importance, list_loss_j, loss_reference = permutation_importance( - X_test, - y_test, + pi = PermutationImportance( estimator=regressor_list[i], n_permutations=50, random_state=0, n_jobs=4, ) - pi_importance_list.append({"importance": importance}) + importance = pi.score(X_test, y_test) + pi_importance_list.append(importance) #############################################################################