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

Exercice

Évaluer la forêt optimale

Dans ce dernier exercice du cours, vous allez évaluer la RMSE sur l'ensemble de test du modèle optimal de grid_rf.

Le jeu de données est déjà chargé et préparé pour vous, avec une division de 80 % pour l'entraînement et 20 % pour le test. Votre environnement contient X_test, y_test et la fonction mean_squared_error de sklearn.metrics sous l'alias MSE. De plus, nous avons aussi chargé l'objet GridSearchCV entraîné, grid_rf, que vous avez instancié à l'exercice précédent. Notez que grid_rf a été entraîné comme suit :

grid_rf.fit(X_train, y_train)

Instructions

100 XP

  • Importez mean_squared_error sous l'alias MSE à partir de sklearn.metrics.

  • Extrayez le meilleur estimateur de grid_rf et assignez-le à best_model.

  • Prédisez les étiquettes de l'ensemble de test avec best_model et assignez le résultat à y_pred.

  • Calculez la RMSE de best_model sur l'ensemble de test.