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

cvičení

Testování CNN modelu pro analýzu sentimentu

Model je teď natrénovaný a PyBooks chce otestovat, jak si poradí s novými recenzemi knih.

Tvým úkolem je zjistit, zda je sentiment v recenzi pozitivní, nebo negativní.

Následující balíčky jsou již naimportované: 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é načtená a uložená jako model.

Pokyny

100 XP
  • Projdi seznam book_reviews a převeď slova v každé recenzi na tensor.
  • Získej výstup modelu pro každý input_tensor.
  • Najdi index nejpravděpodobnější kategorie sentimentu z outputs.data.
  • Extrahuj položku predicted_label a převeď ji na řetězec sentimentu, kde 1 odpovídá štítku "Positive".