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.
Se han precargado las variables X_train
, X_test
, y_train
, y_test
e y_pred
, junto con el modelo ajustado, reg
, todas del último ejercicio.
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 del ejercicio
- Importa
root_mean_squared_error
. - Calcula la puntuación r_squared 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
Prueba este ejercicio completando el código de muestra.
# Import root_mean_squared_error
from ____.____ import ____
# Compute R-squared
r_squared = reg.____(____, ____)
# Compute RMSE
rmse = ____(____, ____)
# Print the metrics
print("R^2: {}".format(____))
print("RMSE: {}".format(____))