1. Nauka
    2. /
  2. Kursy
    3. /
  3. Uczenie głębokie dla tekstu z PyTorch
Connected

ćwiczenie

Budowanie modelu GRU do klasyfikacji tekstu

Zespół PyBooks jest pod wrażeniem wyników dwóch wcześniej wytrenowanych modeli. W dążeniu do doskonałości chce jednak mieć pewność, że wybiera najlepsze możliwe rozwiązanie dla tego zadania. Dlatego poproszono cię o rozszerzenie projektu i sprawdzenie możliwości modeli GRU, cenionych za wydajność i skuteczność w zadaniach klasyfikacji tekstu. Twoim nowym zadaniem jest zastosowanie modelu GRU do klasyfikacji artykułów ze zbioru danych Newsgroup do następujących kategorii:

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

Następujące pakiety zostały już załadowane: torch, nn, optim.

Instrukcje

100 XP
  • Uzupełnij klasę GRU o wymagane parametry.
  • Zainicjalizuj model z tymi samymi parametrami.
  • Wytrenuj model: przekaż parametry do funkcji kryterium i wykonaj propagację wsteczną straty.