Définir des paramètres et ajuster un modèle
Les tâches prédictives entrent dans l’une des deux catégories : régression ou classification. Dans le jeu de données sur les confiseries, la variable cible est une variable continue qui décrit la fréquence à laquelle une confiserie a été choisie plutôt qu’une autre lors d’affrontements en tête-à-tête. Pour prédire cette valeur (le pourcentage de victoires), vous allez utiliser un modèle de régression.
Dans cet exercice, vous allez définir quelques paramètres en utilisant un modèle de régression par forêts aléatoires rfr.
Cet exercice fait partie du cours
Validation des modèles en Python
Instructions
- Ajoutez des paramètres à
rfrpour construire 100 arbres et limiter la profondeur maximale de ces arbres à 6. - Assurez la reproductibilité du modèle en fixant l’état aléatoire à
1111. - Utilisez la méthode
.fit()pour entraîner le modèle de régression par forêts aléatoires avecX_traincomme données d’entrée ety_traincomme variable cible.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Set the number of trees
rfr.____ = ____
# Add a maximum depth
rfr.____ = ____
# Set the random state
rfr.____ = ____
# Fit the model
rfr.____(____, ____)