Eroarea medie pătratică
Să ne întoarcem la predicțiile NBA din 2017. În fiecare an, există cel puțin câteva echipe NBA care câștigă mult mai multe meciuri decât era de așteptat. Dacă folosești MAE, această metrică de acuratețe nu reflectă predicțiile slabe la fel de bine ca MSE. Ridicarea la pătrat a erorilor mari provenite din predicțiile greșite va face ca acuratețea să pară mai slabă.
În acest exemplu, directorii NBA doresc să prezică mai bine numărul de victorii ale echipelor. Vei folosi eroarea medie pătratică pentru a calcula eroarea de predicție. Victoriile reale sunt încărcate ca y_test, iar predicțiile ca predictions.
Acest exercițiu face parte din cursul
Validarea modelelor în Python
Instrucțiuni pentru exercițiu
- Calculează manual MSE. $$ MSE = \frac{\sum_{i=1}^{n} (y_i - \hat{y}_i ) ^2 }{n} $$
- Calculează MSE folosind
sklearn.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
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))