CommencezCommencez gratuitement

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

Voir le cours

Instructions de l’exercice

  • Ajoutez un paramètre à rfr pour 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 avec X_train comme données d'entrée et y_train comme 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.____(____, ____)
Modifier et exécuter le code