Définir des paramètres et entraîner un modèle
Les tâches prédictives appartiennent à l'une de deux catégories : régression ou classification. Dans l'ensemble de données sur les bonbons, la variable cible est continue et décrit la fréquence à laquelle un bonbon a été choisi plutôt qu'un autre lors d'affrontements en tête-à-tête. Pour prédire cette valeur (le pourcentage de victoires), vous utiliserez un modèle de régression.
Dans cet exercice, vous allez préciser quelques paramètres à l'aide d'un modèle de régression par forêts aléatoires rfr.
Cette activité fait partie du cours
Validation de modèles en Python
Instructions de l’exercice
- Ajoutez un paramètre à
rfrpour que le nombre d'arbres construits soit de 100 et que la profondeur maximale de ces arbres soit de 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 de réponse.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Set the number of trees
rfr.____ = ____
# Add a maximum depth
rfr.____ = ____
# Set the random state
rfr.____ = ____
# Fit the model
rfr.____(____, ____)