CommencerCommencer gratuitement

É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

Afficher le cours

Instructions

  • Importez mean_squared_error de sklearn.metrics en tant que MSE.
  • 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))
Modifier et exécuter le code