1. Nauka
    2. /
  2. Kursy
    3. /
  3. Głębokie uczenie z PyTorch – poziom średnio zaawansowany
Connected

ćwiczenie

Ocena modeli prognozowania

Czas na ewaluację! Sieć LSTM, którą trenowałeś/trenowałaś w poprzednim ćwiczeniu, została wytrenowana przez kilka dodatkowych epok i jest dostępna jako net.

Twoim zadaniem jest ocena jej działania na zbiorze testowym przy użyciu metryki błędu średniokwadratowego (torchmetrics zostało już za ciebie zaimportowane). Sprawdźmy, jak dobrze radzi sobie model!

Instrukcje

100 XP
  • Zdefiniuj metrykę błędu średniokwadratowego i przypisz ją do zmiennej mse.
  • Przekaż sekwencję wejściową do net, a wynik przed przypisaniem do outputs wywołaj na nim metodę .squeeze().
  • Oblicz końcową wartość metryki testowej i przypisz ją do test_mse.