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
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))