CommencezCommencez gratuitement

Évaluer le régleur RF

Vous allez maintenant évaluer la RMSE sur l'ensemble de test du régleur random forests rf que vous avez entraîné à l'exercice précédent.

L'ensemble de données a déjà été préparé pour vous et séparé en 80 % entraînement et 20 % test. La matrice de caractéristiques X_test, ainsi que le tableau y_test, sont disponibles dans votre espace de travail. De plus, nous avons aussi chargé le modèle rf que vous avez entraîné à l'exercice précédent.

Cette activité fait partie du cours

Machine Learning avec des modèles à base d'arbres en Python

Voir le cours

Instructions de l’exercice

  • Importez mean_squared_error de sklearn.metrics sous le nom MSE.
  • Prédisez les étiquettes de l'ensemble de test et assignez le résultat à y_pred.
  • Calculez la RMSE sur l'ensemble de test et assignez-la à rmse_test.

Exercice interactif pratique

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

# 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