ComenzarEmpieza gratis

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

Ver curso

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))
Editar y ejecutar código