LoslegenKostenlos loslegen

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 Stimmungs­vorhersage 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

Kurs anzeigen

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}")
Code bearbeiten und ausführen