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
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
yy_pred
. - Imprime
r_squared
yrmse
.
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(____))