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

cvičení

Trénování CNN modelu na textu

Skvělá práce při definování třídy TextClassificationCNN. PyBooks teď potřebuje model natrénovat, aby co nejpřesněji analyzoval sentiment knižních recenzí.

Následující balíčky už jsou za tebe naimportovány: torch, torch.nn jako nn, torch.nn.functional jako F, torch.optim jako optim.

Instance třídy TextClassificationCNN() s argumenty vocab_size a embed_dim je také připravena a uložena jako model.

Pokyny

100 XP
  • Definuj ztrátovou funkci pro binární klasifikaci a ulož ji jako criterion.
  • Na začátku tréninkové smyčky vynuluj gradienty.
  • Na konci smyčky aktualizuj parametry modelu.