CommencerCommencer gratuitement

Formez votre premier arbre de régression

Dans cet exercice, vous allez former un arbre de régression pour prédire la consommation mpg (miles par gallon) des voitures de l' ensemble de données auto-mpg en utilisant les six caractéristiques disponibles.

L'ensemble de données est traité pour vous et divisé en 80 % de formation et 20 % de test. La matrice de caractéristiques X_train et le tableau y_train sont disponibles dans votre espace de travail.

Cet exercice fait partie du cours

Apprentissage automatique avec des modèles arborescents en Python

Afficher le cours

Instructions

  • Importez DecisionTreeRegressor à partir de sklearn.tree.
  • Instantez un site DecisionTreeRegressor dt d'une profondeur maximale de 8 et dont la valeur min_samples_leaf est fixée à 0,13.
  • Ajustez dt à l'ensemble de formation.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____

# Instantiate dt
____ = ____(max_depth=____,
             ____=____,
            random_state=3)

# Fit dt to the training set
____.____(____, ____)
Modifier et exécuter le code