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
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
ey_pred
. - Imprima
r_squared
ermse
.
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(____))