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
Oefeninstructies
Voorspel de labels van de testset met het lineaire regressiemodel (
lr) en sla het resultaat op iny_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))