Update examples/model_evaluation/plot_feature_importance.py

Co-authored-by: Guillaume Lemaitre <[email protected]>
probabl-ai · Mar 3, 2025 · 9e3dd14 · 9e3dd14
1 parent 73928e1
commit 9e3dd14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/model_evaluation/plot_feature_importance.py b/examples/model_evaluation/plot_feature_importance.py
@@ -419,7 +419,7 @@ def sort_absolute_values(df):
 ).set_output(transform="pandas")
 
 parameter_grid = {
-    "columntransformer__kmeans__n_clusters": randint(low=1, high=30),
+    "columntransformer__kmeans__n_clusters": randint(low=10, high=30),
     "columntransformer__remainder__degree": randint(low=1, high=4),
     "columntransformer__remainder__n_knots": randint(low=2, high=10),
     "selectkbest__k": randint(low=5, high=100),