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
Instructions
- Importez
DecisionTreeRegressor
à partir desklearn.tree
. - Instantez un site
DecisionTreeRegressor
dt
d'une profondeur maximale de 8 et dont la valeurmin_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
____.____(____, ____)