CommencezCommencez gratuitement

Évaluer l'ajustement du modèle

Vous allez maintenant mesurer la performance de la régression sur les données d'entraînement et de test avec deux mesures : la racine de l'erreur quadratique moyenne et l'erreur absolue moyenne. C'est une étape essentielle où vous évaluez à quel point les prédictions du modèle sont « proches » des valeurs réelles.

La bibliothèque numpy a été chargée sous le nom np. Les fonctions mean_absolute_error et mean_squared_error ont été chargées. Les variables cibles d'entraînement et de test sont chargées sous train_Y et test_Y, et les valeurs prédites d'entraînement et de test sont importées sous train_pred_Y et test_pred_Y respectivement.

Cette activité fait partie du cours

Machine Learning pour le marketing en Python

Voir le cours

Instructions de l’exercice

  • Calculez la racine de l'erreur quadratique moyenne sur les données d'entraînement en utilisant la fonction np.sqrt().
  • Calculez l'erreur absolue moyenne sur les données d'entraînement.
  • Calculez la racine de l'erreur quadratique moyenne sur les données de test.
  • Calculez l'erreur absolue moyenne sur les données de test.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Calculate root mean squared error on training data
rmse_train = np.sqrt(___(train_Y, train_pred_Y))

# Calculate mean absolute error on training data
mae_train = ___(train_Y, train_pred_Y)

# Calculate root mean squared error on testing data
rmse_test = np.sqrt(___(test_Y, test_pred_Y))

# Calculate mean absolute error on testing data
mae_test = ___(test_Y, test_pred_Y)

# Print the performance metrics
print('RMSE train: {}; RMSE test: {}\nMAE train: {}, MAE test: {}'.format(rmse_train, rmse_test, mae_train, mae_test))
Modifier et exécuter le code