开始使用免费开始使用

评估预测模型

现在进入评估环节!上一练习中您训练的同一个 LSTM 网络,已经又训练了几个 epoch,并以 net 提供给您。

您的任务是使用均方误差(Mean Squared Error)指标在测试数据集上对其进行评估(torchmetrics 已为您导入)。让我们看看模型表现如何!

本练习是课程的一部分

PyTorch 深度学习进阶

查看课程

练习说明

  • 定义均方误差指标并将其赋给 mse
  • 将输入序列传入 net,并在赋给 outputs 之前对结果调用 squeeze
  • 计算测试指标的最终数值,并将其赋给 test_mse

交互式实操练习

通过完成这段示例代码来试试这个练习。

# 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}")
编辑并运行代码