ComenzarEmpieza gratis

Rendimiento de la regresión

Ahora que has ajustado un modelo, reg, utilizando todas las características de sales_df, y has hecho predicciones de los valores de las ventas, puedes evaluar el rendimiento utilizando algunas métricas de regresión habituales.

Las variables X_train, X_test, y_train, y_test, y y_pred, junto con el modelo ajustado, reg, todas del último ejercicio, se han precargado para ti.

Tu tarea consiste en averiguar lo bien que las características pueden explicar la varianza de los valores objetivo, además de evaluar la capacidad del modelo para hacer predicciones sobre datos no vistos.

Este ejercicio forma parte del curso

Aprendizaje supervisado con scikit-learn

Ver curso

Instrucciones de ejercicio

  • Importa mean_squared_error.
  • Calcula la puntuación R-cuadrado del modelo pasando los valores de la característica de prueba y los valores del objetivo de prueba a un método adecuado.
  • Calcula el error cuadrático medio del modelo utilizando y_test y y_pred.
  • Imprime r_squared y rmse.

Ejercicio interactivo práctico

Pruebe este ejercicio completando este código de muestra.

# Import mean_squared_error
from ____.____ import ____

# Compute R-squared
r_squared = reg.____(____, ____)

# Compute RMSE
rmse = ____(____, ____, squared=____)

# Print the metrics
print("R^2: {}".format(____))
print("RMSE: {}".format(____))
Editar y ejecutar código