Entraînez votre premier arbre de régression
Dans cet exercice, vous allez entraîner un arbre de régression pour prédire la consommation mpg (miles per gallon) des voitures dans l'ensemble de données auto-mpg en utilisant les six caractéristiques disponibles.
L'ensemble de données a été préparé pour vous et divisé en 80 % pour l'entraînement et 20 % pour le test. La matrice de caractéristiques X_train et le tableau y_train sont disponibles dans votre espace de travail.
Cette activité fait partie du cours
Machine Learning avec des modèles à base d'arbres en Python
Instructions de l’exercice
- Importez
DecisionTreeRegressordepuissklearn.tree. - Instanciez un
DecisionTreeRegressordtavec une profondeur maximale de 8 etmin_samples_leafréglé à 0.13. - Ajustez
dtà l'ensemble d'entraînement.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____
# Instantiate dt
____ = ____(max_depth=____,
____=____,
random_state=3)
# Fit dt to the training set
____.____(____, ____)