Оценка предобученной модели генерации текста
Команда PyBooks использовала предобученную модель GPT-2, с которой вы уже работали, чтобы сгенерировать текст на основе заданного запроса. Теперь им нужно оценить качество этого текста. Для этого вас попросили сравнить сгенерированный текст с эталонным и вычислить соответствующие метрики.
BLEUScore и ROUGEScore уже загружены для вас.
Это упражнение является частью курса
Глубокое обучение для работы с текстом на PyTorch
Инструкции к упражнению
- Инициализируйте две метрики — BLEU и ROUGE — из модуля
torchmetrics.text. - С помощью инициализированных метрик вычислите оценки для сгенерированного и эталонного текстов.
- Выведите на экран рассчитанные значения метрик BLEU и ROUGE.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
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)