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

cvičení

Vyhodnocení modelu BERT

Teď, když jsou ukázkové recenze tokenizované pomocí BERT tokenizéru, je čas vyhodnotit model BERT na vzorcích z PyBooks. Zároveň otestujeme, jak model předpovídá sentiment na nových datech.

Následující položky jsou již naimportované: BertTokenizer, BertForSequenceClassification, torch. Instance natrénovaného modelu model je také předem načtená. Otestujeme ji na novém datovém vzorku.

Pokyny

100 XP
  • Připrav hodnoticí text pro model tak, že ho tokenizuješ a vrátíš PyTorch tenzory.
  • Převeď výstupní logity na pravděpodobnosti v rozsahu mezi nulou a jednou.
  • Zobraz sentimenty odvozené z pravděpodobností.