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
Instructions
- Importez
DecisionTreeRegressorà partir desklearn.tree. - Instancier un
DecisionTreeRegressordtavec une profondeur maximale de 8 et unmin_samples_leafe fixée à 0,13. - Ajustez l'
dte à 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
____.____(____, ____)