IniziaInizia gratis

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

Visualizza il corso

Istruzioni dell'esercizio

  • Predici le etichette del set di test usando il modello di regressione lineare (lr) e assegna il risultato a y_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))
Modifica ed esegui il codice