É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
Instructions
Importez
mean_squared_error
desklearn.metrics
en tant queMSE
.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))