Aan de slagGa gratis aan de slag

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

Cursus bekijken

Oefeninstructies

  • Importeer de functie mean_squared_error als MSE uit sklearn.metrics.
  • Voorspel de labels van de testset en sla de uitvoer op in y_pred.
  • Bereken de MSE van de testset door MSE aan te roepen en sla het resultaat op in mse_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))
Code bewerken en uitvoeren