Previsão com base no conjunto de teste

No último exercício, aparentemente a regressão linear e a ridge produziram resultados semelhantes. Seria apropriado selecionar qualquer um desses modelos. No entanto, é possível verificar o desempenho preditivo com o conjunto de teste para ver se um deles supera o outro.

Você vai usar como métrica a raiz do erro quadrático médio (RMSE). O dicionário models, que contém os nomes e as instâncias dos dois modelos, foi pré-carregado para você junto com as matrizes de treinamento e da variável dependente X_train_scaled, X_test_scaled, y_train e y_test.

Este exercício faz parte do curso

Aprendizado Supervisionado com o scikit-learn

Ver Curso

Instruções de exercício

  • Importe mean_squared_error.
  • Ajuste o modelo às variáveis independentes de treinamento com escalonamento e aos rótulos de treinamento.
  • Faça previsões usando as variáveis independentes de teste com escalonamento.
  • Calcule o RMSE passando os rótulos do conjunto de teste e os rótulos previstos.

Exercício interativo prático

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

# Import mean_squared_error
from ____.____ import ____

for name, model in models.items():
  
  # Fit the model to the training data
  ____
  
  # Make predictions on the test set
  y_pred = ____
  
  # Calculate the test_rmse
  test_rmse = ____(____, ____, squared=____)
  print("{} Test Set RMSE: {}".format(name, test_rmse))