Erreur quadratique moyenne
Revenons aux prédictions de la NBA pour 2017. Chaque année, il y a au moins quelques équipes de la NBA qui remportent bien plus de matchs que prévu. Si vous utilisez la MAE, cette mesure de justesse reflète moins les mauvaises prédictions que si vous utilisez la MSE. Le fait d'élever au carré les grandes erreurs issues de mauvaises prédictions fera paraître la justesse plus faible.
Dans cet exemple, les dirigeants de la NBA veulent mieux prédire le nombre de victoires des équipes. Vous utiliserez 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.
Cette activité fait partie du cours
Validation de modèles en Python
Instructions de l’exercice
- 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 ce code d’exemple.
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))