Allena il tuo primo albero di regressione
In questo esercizio allenerai un albero di regressione per prevedere il consumo mpg (miglia per gallone) delle auto nel dataset auto-mpg utilizzando tutte e sei le feature disponibili.
Il dataset è già stato elaborato per te ed è suddiviso in 80% train e 20% test. La matrice delle feature X_train e l'array y_train sono disponibili nel tuo workspace.
Questo esercizio fa parte del corso
Machine Learning con modelli ad alberi in Python
Istruzioni dell'esercizio
- Importa
DecisionTreeRegressordasklearn.tree. - Istanzia un
DecisionTreeRegressordtcon profondità massima pari a 8 emin_samples_leafimpostato a 0.13. - Esegui il fit di
dtsul training set.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____
# Instantiate dt
____ = ____(max_depth=____,
____=____,
random_state=3)
# Fit dt to the training set
____.____(____, ____)