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
Istruzioni dell'esercizio
- Importa
train_test_splitdasklearn.model_selection. - Suddividi i dati in 70% train e 30% test.
- Istanzia un
DecisionTreeRegressorcon profondità massima pari a 4 emin_samples_leafimpostato 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)