Utvärdera GB-regressorn
Nu när prediktionerna för testmängden är tillgängliga kan du använda dem för att beräkna RMSE (Root Mean Squared Error) för testmängden med gb.
y_test och prediktionerna y_pred finns tillgängliga i din arbetsyta.
Den här övningen är en del av kursen
Maskininlärning med trädbaserade modeller i Python
Övningsinstruktioner
Importera
mean_squared_errorfrånsklearn.metricssomMSE.Beräkna MSE för testmängden och tilldela resultatet till
mse_test.Beräkna RMSE för testmängden och tilldela resultatet till
rmse_test.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Import mean_squared_error as MSE
____
# Compute MSE
mse_test = ____
# Compute RMSE
rmse_test = ____
# Print RMSE
print('Test set RMSE of gb: {:.3f}'.format(rmse_test))