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() | |