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
Instrucciones de ejercicio
- Importa
mean_squared_error
desklearn.metrics
comoMSE
. - 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))