ComeçarComece de graça

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

Ver curso

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 a outputs.
  • 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}")
Editar e executar o código