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 ( 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.
  • Instancier un DecisionTreeRegressor dt avec une profondeur maximale de 8 et un min_samples_leaf e fixée à 0,13.
  • Ajustez l'dt e à 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