1. Apprendre
    2. /
  2. Cours
    3. /
  3. Machine Learning avec des modèles à base d'arbres en Python
Connected

Exercice

É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.

Instructions

100 XP
  • 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.