1. Learn
    2. /
  2. คอร์ส
    3. /
  3. Uczenie głębokie dla tekstu z PyTorch
Connected

แบบฝึกหัด

Trenowanie modelu CNN do klasyfikacji tekstu

Świetna robota z definiowaniem klasy TextClassificationCNN. PyBooks musi teraz wytrenować model, aby zoptymalizować go pod kątem trafnej analizy sentymentu recenzji książek.

Następujące pakiety zostały już zaimportowane: torch, torch.nn jako nn, torch.nn.functional jako F, torch.optim jako optim.

Instancja klasy TextClassificationCNN() z argumentami vocab_size i embed_dim została również załadowana i zapisana jako model.

คำแนะนำ

100 XP
  • Zdefiniuj funkcję straty używaną do klasyfikacji binarnej i zapisz ją jako criterion.
  • Wyzeruj gradienty na początku pętli treningowej.
  • Zaktualizuj parametry na końcu pętli.