Evaluate the RF regressor
You'll now evaluate the test set RMSE of the random forests regressor rf
that you trained in the previous exercise.
The dataset is processed for you and split into 80% train and 20% test. The features matrix X_test
, as well as the array y_test
are available in your workspace. In addition, we have also loaded the model rf
that you trained in the previous exercise.
This exercise is part of the course
Machine Learning with Tree-Based Models in Python
Exercise instructions
- Import
mean_squared_error
fromsklearn.metrics
asMSE
. - Predict the test set labels and assign the result to
y_pred
. - Compute the test set RMSE and assign it to
rmse_test
.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# 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))