Inizia subitoInizia gratis

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

Visualizza corso

Istruzioni dell'esercizio

  • Importa DecisionTreeRegressor da sklearn.tree.
  • Istanzia un DecisionTreeRegressor dt con profondità massima pari a 8 e min_samples_leaf impostato a 0.13.
  • Esegui il fit di dt sul training set.

esercizio interattivo pratico

Prova questo esercizio completando questo codice di esempio.

# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____

# Instantiate dt
____ = ____(max_depth=____,
             ____=____,
            random_state=3)

# Fit dt to the training set
____.____(____, ____)
Modifica ed esegui il codice