1. Learn
    2. /
  2. 课程
    3. /
  3. PyTorch 深度学习进阶
Connected

道练习

评估预测模型

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

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

说明

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