Evalueer de regressieboom
In deze oefening evalueer je de prestaties van dt op de testset met de Root Mean Squared Error (RMSE). De RMSE van een model meet gemiddeld hoeveel de voorspellingen van het model afwijken van de echte labels. Je krijgt de RMSE door de vierkantswortel te nemen van de Mean Squared Error (MSE) van het model.
De featurematrix X_test, de array y_test en de decision tree regressor dt die je in de vorige oefening hebt getraind, zijn beschikbaar in je werkruimte.
Deze oefening maakt deel uit van de cursus
Machine Learning met boomgebaseerde modellen in Python
Oefeninstructies
- Importeer de functie
mean_squared_erroralsMSEuitsklearn.metrics. - Voorspel de labels van de testset en sla de uitvoer op in
y_pred. - Bereken de MSE van de testset door
MSEaan te roepen en sla het resultaat op inmse_dt. - Bereken de RMSE van de testset en sla deze op in
rmse_dt.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Import mean_squared_error from sklearn.metrics as MSE
from ____.____ import ____ as ____
# Compute y_pred
____ = ____.____(____)
# Compute mse_dt
____ = ____(____, ____)
# Compute rmse_dt
____ = ____
# Print rmse_dt
print("Test set RMSE of dt: {:.2f}".format(rmse_dt))