CommencerCommencer gratuitement

Entraînez votre premier arbre de régression

Dans cet exercice, vous allez entraîner un arbre de régression afin de prédire la consommation de carburant (en mpg) des voitures dans 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 deux parties : 80 % pour l'entraînement et 20 % pour le test. La matrice des fonctionnalités X_train et le tableau y_train sont disponibles dans votre espace de travail.

Cet exercice fait partie du cours

Machine learning avec des modèles arborescents en Python

Afficher le cours

Instructions

  • Importez DecisionTreeRegressor à partir de sklearn.tree.
  • Instanciez un DecisionTreeRegressor dt avec une profondeur maximale de 8 et un min_samples_leaf fixé à 0,13.
  • Ajustez dt à l'ensemble d'apprentissage.

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