Error cuadrático medio
Volvamos a centrarnos en las predicciones de la NBA de 2017. Cada año, hay al menos un par de equipos de la NBA que ganan muchísimos más partidos de lo esperado. Si usas el MAE, esta métrica de exactitud no refleja tan claramente las malas predicciones como lo hace el MSE. Elevar al cuadrado los errores grandes de malas predicciones hará que la exactitud parezca peor.
En este ejemplo, los directivos de la NBA quieren predecir mejor las victorias de los equipos. Utilizarás el error cuadrático medio para calcular el error de predicción. Las victorias reales están cargadas como y_test y las predicciones como predictions.
Este ejercicio forma parte del curso
Validación de modelos en Python
Instrucciones del ejercicio
- Calcula manualmente el MSE. $$ MSE = \frac{\sum_{i=1}^{n} (y_i - \hat{y}_i ) ^2 }{n} $$
- Calcula el MSE usando
sklearn.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
from sklearn.metrics import ____
n = ___(predictions)
# Finish the manual calculation of the MSE
mse_one = sum((y_test - predictions)____) / n
print('With a manual calculation, the error is {}'.format(mse_one))
# Use the scikit-learn function to calculate MSE
mse_two = ____
print('Using scikit-learn, the error is {}'.format(mse_two))