1. 학습
  2. /
  3. 강의
  4. /
  5. PyTorchによる中級ディープラーニング

Connected

연습 문제

予測モデルの評価

評価の時間です!前の演習で学習したのと同じ LSTM ネットワークを、さらに数エポック学習させたものが net として用意されています。

あなたのタスクは、平均二乗誤差(Mean Squared Error)指標を使ってテストデータセットで評価することです(torchmetrics はすでにインポート済みです)。モデルの性能を確認してみましょう!

지침

100 XP
  • 平均二乗誤差(Mean Squared Error)指標を定義して、mse に代入します。
  • 入力シーケンスを net に渡し、結果を .squeeze() してから outputs に代入します。
  • テスト指標の最終値を計算し、test_mse に代入します。