ComenzarEmpieza gratis

Evaluar el bosque óptimo

En este último ejercicio del curso, evaluarás el conjunto de pruebas RMSE del modelo óptimo de grid_rf.

El conjunto de datos ya está cargado y procesado para ti, y está dividido en un 80 % de entrenamiento y un 20 % de prueba. En tu entorno están disponibles X_test, y_test y la función mean_squared_error de sklearn.metrics bajo el alias MSE. Además, también hemos cargado el objeto GridSearchCV entrenado grid_rf que instanciaste en el ejercicio anterior. Ten en cuenta que grid_rf se entrenó del siguiente modo:

grid_rf.fit(X_train, y_train)

Este ejercicio forma parte del curso

Machine learning con modelos basados en árboles en Python

Ver curso

Instrucciones de ejercicio

  • Importa mean_squared_error como MSE desde sklearn.metrics.

  • Extrae el mejor estimador de grid_rf y asígnalo a best_model.

  • Predice las etiquetas del conjunto de pruebas de best_model y asigna el resultado a y_pred.

  • Calcula el conjunto de pruebas best_model's RMSE.

Ejercicio interactivo práctico

Pruebe este ejercicio completando este código de muestra.

# Import mean_squared_error from sklearn.metrics as MSE 
____

# Extract the best estimator
best_model = ____

# Predict test set labels
y_pred = ____

# Compute rmse_test
rmse_test = ____

# Print rmse_test
print('Test RMSE of best model: {:.3f}'.format(rmse_test)) 
Editar y ejecutar código