Avaliar o regressor RF
Agora você avaliará o conjunto de testes RMSE do regressor de florestas aleatórias rf
que você treinou no exercício anterior.
O conjunto de dados é processado para você e dividido em 80% de treinamento e 20% de teste. A matriz de recursos X_test
, bem como a matriz y_test
estão disponíveis em seu espaço de trabalho. Além disso, também carregamos o modelo rf
que você treinou no exercício anterior.
Este exercício faz parte do curso
Aprendizado de máquina com modelos baseados em árvores em Python
Instruções de exercício
- Importe
mean_squared_error
desklearn.metrics
comoMSE
. - Preveja os rótulos do conjunto de teste e atribua o resultado a
y_pred
. - Calcule o conjunto de teste RMSE e atribua-o a
rmse_test
.
Exercício interativo prático
Experimente este exercício preenchendo este código de exemplo.
# 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))