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