Evalúa el regresor GB
Ahora que dispones de las predicciones del conjunto de pruebas, puedes utilizarlas para evaluar el error cuadrático medio del conjunto de pruebas (RMSE) de gb
.
y_test
y predicciones y_pred
están disponibles en tu espacio de trabajo.
Este ejercicio forma parte del curso
Machine learning con modelos basados en árboles en Python
Instrucciones del ejercicio
Importa
mean_squared_error
desklearn.metrics
comoMSE
.Calcula el conjunto de prueba MSE y asígnalo a
mse_test
.Calcula el conjunto de prueba RMSE y asígnalo a
rmse_test
.
Ejercicio interactivo práctico
Prueba este ejercicio completando el código de muestra.
# Import mean_squared_error as MSE
____
# Compute MSE
mse_test = ____
# Compute RMSE
rmse_test = ____
# Print RMSE
print('Test set RMSE of gb: {:.3f}'.format(rmse_test))