IniziaInizia 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 il 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 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
____.____(____, ____)
Modifica ed esegui il codice