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

ćwiczenie

Ewaluacja modelu BERT

Po stokenizowaniu przykładowych recenzji za pomocą tokenizatora BERT nadszedł czas na ewaluację modelu BERT na próbkach z PyBooks. Ocenisz też predykcję sentymentu modelu na nowych danych.

Następujące elementy zostały już zaimportowane: BertTokenizer, BertForSequenceClassification, torch. Wytrenowana instancja model jest również wstępnie załadowana. Przetestujesz ją teraz na nowej próbce danych.

Instrukcje

100 XP
  • Przygotuj tekst do ewaluacji, tokenizując go i zwracając tensory PyTorch.
  • Przekształć wyjściowe logity na prawdopodobieństwa z zakresu od zera do jedności.
  • Wyświetl sentyment na podstawie uzyskanych prawdopodobieństw.