Évaluer la forêt optimale
Dans le dernier exercice du cours, vous évaluerez la RMSE du modèle optimal de grid_rf.
L'ensemble de données est déjà chargé et traité pour vous, et est divisé en 80 % d'entraînement et 20 % de test. Dans votre environnement, les éléments suivants sont disponibles : X_test, y_test et la fonction mean_squared_error de sklearn.metrics sous l'alias MSE. De plus, nous avons également chargé l'objet GridSearchCV formé grid_rf que vous avez instancié dans l'exercice précédent. Notez que grid_rf a été formé de la manière suivante :
grid_rf.fit(X_train, y_train)
Cet exercice fait partie du cours
Machine learning avec des modèles arborescents en Python
Instructions
Importez
mean_squared_erroren tant queMSEdepuissklearn.metrics.Veuillez extraire le meilleur estimateur de
grid_rfet l'attribuer àbest_model.Veuillez prédire les étiquettes de l'ensemble de test
best_modelet attribuer le résultat ày_pred.Calculez la RMSE de l'ensemble de test de
best_model.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Import mean_squared_error from sklearn.metrics as MSE
____
# Extract the best estimator
best_model = ____
# Predict test set labels
y_pred = ____
# Compute rmse_test
rmse_test = ____
# Print rmse_test
print('Test RMSE of best model: {:.3f}'.format(rmse_test))