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
Instructions
- Importez
DecisionTreeRegressorà partir desklearn.tree. - Instanciez un
DecisionTreeRegressordtavec une profondeur maximale de 8 et unmin_samples_leaffixé à 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
____.____(____, ____)