CommencerCommencer gratuitement

Évaluez le régresseur GB

Maintenant que les prédictions de l'ensemble de test sont disponibles, vous pouvez les utiliser pour évaluer l'erreur quadratique moyenne de l'ensemble de test (RMSE) de gb.

y_test et les prédictions y_pred sont disponibles dans votre espace de travail.

Cet exercice fait partie du cours

Apprentissage automatique avec des modèles arborescents en Python

Afficher le cours

Instructions

  • Importez mean_squared_error de sklearn.metrics en tant que MSE.

  • Calculez l'ensemble de test MSE et affectez-le à mse_test.

  • Calculez l'ensemble de test RMSE et affectez-le à rmse_test.

Exercice interactif pratique

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

# Import mean_squared_error as MSE
____

# Compute MSE
mse_test = ____

# Compute RMSE
rmse_test = ____

# Print RMSE
print('Test set RMSE of gb: {:.3f}'.format(rmse_test))
Modifier et exécuter le code