CommencerCommencer gratuitement

Erreur quadratique moyenne

Revenons aux prédictions NBA de 2017. Chaque année, il y a au moins deux équipes NBA qui gagnent bien plus de matchs que prévu. Si vous utilisez la MAE, cette métrique de précision reflète moins les mauvaises prédictions que si vous utilisez la MSE. En élevant au carré les grandes erreurs, la précision paraîtra plus faible.

Dans cet exemple, les dirigeants de la NBA souhaitent mieux prédire le nombre de victoires des équipes. Vous allez utiliser l'erreur quadratique moyenne pour calculer l'erreur de prédiction. Les victoires réelles sont chargées dans y_test et les prédictions dans predictions.

Cet exercice fait partie du cours

Validation des modèles en Python

Afficher le cours

Instructions

  • Calculez manuellement la MSE. $$ MSE = \frac{\sum_{i=1}^{n} (y_i - \hat{y}_i ) ^2 }{n} $$
  • Calculez la MSE avec sklearn.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

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))
Modifier et exécuter le code