ComeçarComece de graça

Desempenho da regressão

Agora que você ajustou um modelo, reg, usando todas as variáveis independentes de sales_df e fez previsões de valores de vendas, pode avaliar o desempenho usando algumas métricas de regressão comuns.

As variáveis X_train, X_test, y_train, y_test e y_pred, juntamente com o modelo ajustado, reg, todos do último exercício, foram pré-carregados para você.

Sua tarefa é descobrir até que ponto as variáveis independentes conseguem explicar a variância dos valores da variável dependente, além de avaliar a capacidade do modelo de fazer previsões sobre dados não vistos.

Este exercício faz parte do curso

Aprendizado Supervisionado com o scikit-learn

Ver curso

Instruções do exercício

  • Importe root_mean_squared_error.
  • Calcule a pontuação R-quadrado do modelo passando os valores das variáveis independentes de teste e os valores da variável dependente de teste em um método apropriado.
  • Calcule a raiz do erro quadrático médio do modelo usando y_test e y_pred.
  • Imprima r_squared e rmse.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

# Import root_mean_squared_error
from ____.____ import ____

# Compute R-squared
r_squared = reg.____(____, ____)

# Compute RMSE
rmse = ____(____, ____)

# Print the metrics
print("R^2: {}".format(____))
print("RMSE: {}".format(____))
Editar e executar o código