1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Intermediate Deep Learning with PyTorch
Connected

cvičení

Vyhodnocení predikcí modelu

Čas na vyhodnocení! Stejná LSTM síť, kterou jsi trénoval/a v předchozím cvičení, byla pro tebe dotrénována na několik dalších epoch a je dostupná jako net.

Tvým úkolem je otestovat ji na testovací datové sadě pomocí metriky Mean Squared Error (knihovna torchmetrics je již naimportovaná). Pojďme se podívat, jak si model vede!

Pokyny

100 XP
  • Definuj metriku Mean Squared Error a přiřaď ji do proměnné mse.
  • Předej vstupní sekvenci do net a výsledek před přiřazením do outputs uprav pomocí .squeeze().
  • Vypočítej výslednou hodnotu testovací metriky a přiřaď ji do test_mse.