MCQ-Rake / testers /rouge-tester.py
mikymatt's picture
feat: release
98dc5b0
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()