CommencerCommencer gratuitement

Évaluer le régresseur RF

Vous allez maintenant évaluer l'erreur quadratique moyenne (RMSE) de l'ensemble de test du régresseur Random Forests rf que vous avez formé dans l'exercice précédent.

L'ensemble de données est traité pour vous et divisé en 80 % d'entraînement et 20 % de test. La matrice des fonctionnalités X_test ainsi que le tableau y_test sont disponibles dans votre espace de travail. De plus, nous avons également chargé le modèle rf que vous avez formé dans l'exercice précédent.

Cet exercice fait partie du cours

Machine learning avec des modèles arborescents en Python

Afficher le cours

Instructions

  • Veuillez importer mean_squared_error depuis sklearn.metrics en tant que MSE.
  • Veuillez prédire les étiquettes de l'ensemble de test et attribuer le résultat à y_pred.
  • Calculez la RMSE de l'ensemble de test et attribuez-la à 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))
Modifier et exécuter le code