Évaluez le régresseur RF
Vous allez maintenant évaluer l'ensemble de test RMSE du régresseur de forêts aléatoires rf
que vous avez formé dans l'exercice précédent.
L'ensemble de données est traité pour vous et divisé en 80 % de formation et 20 % de test. La matrice de caractéristiques X_test
, ainsi que le tableau y_test
sont disponibles dans votre espace de travail. En outre, nous avons également chargé le modèle rf
que vous avez formé dans l'exercice précédent.
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
. - Prédire les étiquettes de l'ensemble de test et attribuer le résultat à
y_pred
. - 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
from ____.____ import ____ as ____
# Predict the test set labels
y_pred = ____.____(____)
# Evaluate the test set RMSE
rmse_test = ____
# Print rmse_test
print('Test set RMSE of rf: {:.2f}'.format(rmse_test))