Spaces:
Runtime error
Runtime error
File size: 888 Bytes
8eab63f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import evaluate
def evaluate_summaries(predictions, references):
"""
Evaluates generated summaries against reference summaries using ROUGE, BLEU and METEOR.
Args:
predictions (list): A list of generated summaries (strings).
references (list): A list of reference summaries (strings).
Returns:
dict: A dictionary containing the ROUGE scores.
"""
rouge = evaluate.load("rouge")
bleu = evaluate.load("bleu")
meteor = evaluate.load("meteor")
rouge_results = rouge.compute(predictions=predictions, references=references)
bleu_results = bleu.compute(predictions=predictions, references=references)
meteor_results = meteor.compute(predictions=predictions, references=references)
return {
"rouge": rouge_results,
"bleu": bleu_results,
"meteor": meteor_results,
}
|