Avaliação de modelos de previsão
É hora da avaliação! A mesma rede LSTM que você treinou no exercício anterior foi treinada para você por mais algumas épocas e está disponível em net
.
Sua tarefa é avaliá-lo em um conjunto de dados de teste usando a métrica Mean Squared Error (torchmetrics
já foi importado para você). Vamos ver se o modelo está indo bem!
Este exercício faz parte do curso
Aprendizagem profunda intermediária com PyTorch
Instruções do exercício
- Defina a métrica de erro quadrático médio e atribua-a a
mse
. - Passe a sequência de entrada para
net
e comprima o resultado antes de você atribuí-lo aoutputs
. - Calcule o valor final da métrica de teste atribuindo-o a
test_mse
.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Define MSE metric
mse = ____
net.eval()
with torch.no_grad():
for seqs, labels in dataloader_test:
seqs = seqs.view(32, 96, 1)
# Pass seqs to net and squeeze the result
outputs = ____
mse(outputs, labels)
# Compute final metric value
test_mse = ____
print(f"Test MSE: {test_mse}")