Aan de slagGa gratis aan de slag

Evalueer de RF-regressor

Je gaat nu de RMSE op de testset evalueren van de random forests-regressor rf die je in de vorige oefening hebt getraind.

De gegevensset is alvast voor je verwerkt en gesplitst in 80% train en 20% test. De featurematrix X_test en de array y_test staan voor je klaar in je werkruimte. Daarnaast hebben we ook het model rf geladen dat je in de vorige oefening hebt getraind.

Deze oefening maakt deel uit van de cursus

Machine Learning met boomgebaseerde modellen in Python

Cursus bekijken

Oefeninstructies

  • Importeer mean_squared_error uit sklearn.metrics als MSE.
  • Voorspel de labels voor de testset en wijs het resultaat toe aan y_pred.
  • Bereken de RMSE op de testset en wijs deze toe aan rmse_test.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# Import mean_squared_error as MSE
from ____.____ import ____ as ____

# Predict the test set labels
y_pred = ____.____(____)

# Evaluate the test set RMSE
rmse_test = ____

# Print rmse_test
print('Test set RMSE of rf: {:.2f}'.format(rmse_test))
Code bewerken en uitvoeren