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

Ver Curso

Instruções de exercício

  • Importe mean_squared_error de sklearn.metrics como MSE.
  • 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))