Das BERT-Modell auswerten
Nachdem du die Beispielrezensionen mit dem BERT-Tokenizer tokenisiert hast, ist es nun Zeit, das BERT-Modell mit den PyBooks-Beispielen auszuwerten. Zusätzlich überprüfst du die Stimmungsvorhersage des Modells an neuen Daten.
Folgendes wurde für dich importiert: BertTokenizer, BertForSequenceClassification, torch.
Die trainierte model-Instanz ist ebenfalls vorab geladen. Wir testen sie jetzt an einem neuen Datenbeispiel.
Diese Übung ist Teil des Kurses
Deep Learning für Text mit PyTorch
Anleitung zur Übung
- Bereite den Bewertungstext für das Modell vor, indem du ihn tokenisierst und PyTorch-Tensoren zurückgibst.
- Wandle die Ausgabe-Logits in Wahrscheinlichkeiten zwischen null und eins um.
- Zeige die Stimmungen basierend auf den Wahrscheinlichkeiten an.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
text = "I had an awesome day!"
# Tokenize the text and return PyTorch tensors
input_eval = tokenizer(____, return_tensors=____, truncation=True, padding=True, max_length=32)
outputs_eval = model(**input_eval)
# Convert the output logits to probabilities
predictions = torch.nn.functional.____(outputs_eval.____, dim=-1)
# Display the sentiments
predicted_label = ____ if torch.____(predictions) > 0 else ____
print(f"Text: {text}\nSentiment: {predicted_label}")