CommencerCommencer gratuitement

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

Afficher le cours

Instructions

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