Evalúa el regresor RF

Ahora evaluarás el conjunto de prueba RMSE del regresor de bosques aleatorios rf que entrenaste en el ejercicio anterior.

El conjunto de datos se procesa por ti y se divide en un 80 % de entrenamiento y un 20 % de prueba. La matriz de características X_test, así como la matriz y_test están disponibles en tu espacio de trabajo. Además, también hemos cargado el modelo rf que entrenaste en el ejercicio anterior.

Este ejercicio forma parte del curso

Machine learning con modelos basados en árboles en Python

Ver curso

Instrucciones de ejercicio

  • Importa mean_squared_error de sklearn.metrics como MSE.
  • Predice las etiquetas del conjunto de pruebas y asigna el resultado a y_pred.
  • Calcula el conjunto de prueba RMSE y asígnalo a rmse_test.

Ejercicio interactivo práctico

Pruebe este ejercicio completando este código de muestra.

# 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))