CommencerCommencez gratuitement

Évaluer 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 (RMSE) de l'ensemble de test de gb.

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

Cet exercice fait partie du cours

<cours>Machine learning avec des modèles arborescents en Python</cours>
Voir le cours

Instructions de l’exercice

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

  • Calculez la MSE de l'ensemble de test et attribuez-la à mse_test.

  • Calculez la RMSE de l'ensemble de test et attribuez-la à rmse_test.

Exercice interactif pratique

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

# 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