1. Learn
    2. /
  2. คอร์ส
    3. /
  3. Deep Learning for Text with PyTorch
Connected

แบบฝึกหัด

Sestavení modelu GRU pro text

Tým v PyBooks byl nadšený výsledky dvou modelů, které jsi předtím natrénoval/a. V touze dosáhnout co nejlepších výsledků ale chtějí mít jistotu, že pro daný úkol vyberou skutečně nejlepší model. Požádali tě proto o rozšíření projektu – konkrétně o vyzkoušení možností modelů GRU, které jsou známé svou efektivitou a spolehlivostí v úlohách klasifikace textu. Tvým novým úkolem je použít model GRU ke klasifikaci článků z datasetu Newsgroup do následujících kategorií:

rec.autos, sci.med a comp.graphics.

Následující balíčky už jsou načteny: torch, nn, optim.

คำแนะนำ

100 XP
  • Dokonči třídu GRU s požadovanými parametry.
  • Inicializuj model se stejnými parametry.
  • Natrénuj model: předej parametry funkci criterion a proveď zpětnou propagaci chyby.