IniziaInizia gratis

Istanzia il modello

Nella seguente serie di esercizi, diagnosticherai i problemi di bias e variance di un albero di regressione. L’albero di regressione che definirai in questo esercizio verrà usato per prevedere il consumo mpg delle auto dal dataset auto utilizzando tutte le feature disponibili.

Abbiamo già preprocessato i dati e caricato la matrice delle feature X e l’array y nel tuo workspace. Inoltre, la classe DecisionTreeRegressor è stata importata da sklearn.tree.

Questo esercizio fa parte del corso

Machine Learning con modelli ad alberi in Python

Visualizza il corso

Istruzioni dell'esercizio

  • Importa train_test_split da sklearn.model_selection.
  • Suddividi i dati in 70% train e 30% test.
  • Istanzia un DecisionTreeRegressor con profondità massima pari a 4 e min_samples_leaf impostato a 0.26.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

# Import train_test_split from sklearn.model_selection
____

# Set SEED for reproducibility
SEED = 1

# Split the data into 70% train and 30% test
X_train, X_test, y_train, y_test = ____(____, ____, test_size=____, random_state=SEED)

# Instantiate a DecisionTreeRegressor dt
dt = ____(____=____, ____=____, random_state=SEED)
Modifica ed esegui il codice