Kom igångKom igång gratis

Utvärdera en förtränad textgenereringsmodell

PyBooks-teamet har använt en förtränad GPT-2-modell som du experimenterade med för att generera text utifrån en given prompt. Nu vill de utvärdera kvaliteten på den genererade texten. Din uppgift är att utvärdera den genererade texten med hjälp av en referenstext.

BLEUScore, ROUGEScore har lästs in åt dig.

Den här övningen är en del av kursen

Djupinlärning för text med PyTorch

Visa kurs

Övningsinstruktioner

  • Börja med att initialisera de två måtten (BLEU och ROUGE) från torchmetrics.text.
  • Använd de initialiserade måtten för att beräkna poängen mellan den genererade texten och referenstexten.
  • Visa de beräknade BLEU- och ROUGE-poängen.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

reference_text = "Once upon a time, there was a little girl who lived in a village near the forest."
generated_text = "Once upon a time, the world was a place of great beauty and great danger. The world of the gods was the place where the great gods were born, and where they were to live."

# Initialize BLEU and ROUGE scorers
bleu = ____()
rouge = ____()

# Calculate the BLEU and ROUGE scores
bleu_score = bleu([____], [[reference_text]])
rouge_score = rouge([generated_text], [[____]])

# Print the BLEU and ROUGE scores
print("BLEU Score:", bleu_score.____())
print("ROUGE Score:", rouge_score)
Redigera och kör kod