File size: 522 Bytes
98dc5b0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from rouge import Rouge
hypothesis = "this is a dog"
reference = "this is a dog"
rouge = Rouge()
scores = rouge.get_scores(hypothesis, reference, avg=True)
for rouge_type in scores.keys():
print(rouge_type)
for score in scores[rouge_type]:
if(score == 'r'):
print(f"recall: {scores[rouge_type][score]}")
if(score == 'p'):
print(f"precision: {scores[rouge_type][score]}")
if(score == 'f'):
print(f"f1_score: {scores[rouge_type][score]}")
print()
|