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

cvičení

Sestavení modelu LSTM pro text

Tým v PyBooks neustále hledá způsoby, jak zlepšit uživatelský zážitek díky nejnovějším technologiím. V souladu s touto vizí ti byl přidělen důležitý úkol. Tým chce prozkoumat možnosti dalšího výkonného nástroje: LSTM, který dokáže zachytit složitější vzory v datech. Pracuješ se stejnou datovou sadou Newsgroup a cíl zůstává stejný: klasifikovat novinové články do tří různých kategorií:

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

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

Pokyny

100 XP
  • Nastav model LSTM tak, že doplníš vrstvu LSTM a lineární vrstvu s potřebnými parametry.
  • Inicializuj model s potřebnými parametry.
  • Natrénuj model LSTM: vynuluj gradienty a předej vstupní data X_train_seq modelu.
  • Vypočítej ztrátu na základě predikovaných hodnot outputs a skutečných štítků.