Regressione lineare vs albero di regressione
In questo esercizio confronterai l’RMSE sul set di test di dt con quello ottenuto da un modello di regressione lineare. Abbiamo già istanziato un modello di regressione lineare lr e lo abbiamo addestrato sullo stesso insieme di dati di dt.
La matrice delle caratteristiche X_test, l’array delle etichette y_test, il modello di regressione lineare addestrato lr, la funzione mean_squared_error importata con l’alias MSE e rmse_dt dall’esercizio precedente sono disponibili nel tuo workspace.
Questo esercizio fa parte del corso
Machine Learning con modelli ad alberi in Python
Istruzioni dell'esercizio
Predici le etichette del set di test usando il modello di regressione lineare (
lr) e assegna il risultato ay_pred_lr.Calcola l’MSE sul set di test e assegna il risultato a
mse_lr.Calcola l’RMSE sul set di test e assegna il risultato a
rmse_lr.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# Predict test set labels
____ = ____.____(____)
# Compute mse_lr
____ = ____(____, ____)
# Compute rmse_lr
____ = ____
# Print rmse_lr
print('Linear Regression test set RMSE: {:.2f}'.format(rmse_lr))
# Print rmse_dt
print('Regression Tree test set RMSE: {:.2f}'.format(rmse_dt))