CommencezCommencez gratuitement

Ajuster d'autres hyperparamètres

La vraie puissance de GridSearchCV se manifeste lorsque vous ajustez plusieurs hyperparamètres : l'algorithme essaie alors toutes les combinaisons possibles pour trouver la meilleure. Ici, vous allez ajuster les hyperparamètres suivants d'une forêt aléatoire :

Hyperparamètre Rôle
criterion Qualité de la division
max_features Nombre de variables pour la meilleure division
max_depth Profondeur maximale de l'arbre
bootstrap Indique si des échantillons Bootstrap sont utilisés

La grille d'hyperparamètres a été définie pour vous, ainsi qu'un classificateur de forêt aléatoire nommé clf.

Cette activité fait partie du cours

Analytique marketing : prédire l'attrition de la clientèle avec Python

Voir le cours

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Import GridSearchCV
from sklearn.model_selection import GridSearchCV

# Create the hyperparameter grid
param_grid = {"max_depth": [3, None],
              "max_features": [1, 3, 10],
              "bootstrap": [True, False],
              "criterion": ["gini", "entropy"]}

# Call GridSearchCV
grid_search = ____(___,___,cv=3)
Modifier et exécuter le code