Aan de slagGa gratis aan de slag

Lineaire regressie vs regressieboom

In deze oefening vergelijk je de RMSE op de testset van dt met die van een lineair regressiemodel. We hebben al een lineair regressiemodel lr aangemaakt en op dezelfde gegevensset getraind als dt.

De feature-matrix X_test, de array met labels y_test, het getrainde lineaire regressiemodel lr, de functie mean_squared_error die is geïmporteerd met de alias MSE, en rmse_dt uit de vorige oefening zijn beschikbaar in je werkruimte.

Deze oefening maakt deel uit van de cursus

Machine Learning met boomgebaseerde modellen in Python

Cursus bekijken

Oefeninstructies

  • Voorspel de labels van de testset met het lineaire regressiemodel (lr) en sla het resultaat op in y_pred_lr.

  • Bereken de MSE op de testset en sla het resultaat op in mse_lr.

  • Bereken de RMSE op de testset en sla het resultaat op in rmse_lr.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# 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))
Code bewerken en uitvoeren