def rouge_score()

in utils/compute_score.py [0:0]

• 9 lines of code
• 1 McCabe index (conditional complexity)

def rouge_score(reference, candidate):
    rouge_calculator = Rouge()
    # Rouge-1 [r, p, f]
    rouge_1 = rouge_calculator.get_scores(candidate, reference, avg=True)['rouge-1']['f']

    # Rouge-2
    rouge_2 = rouge_calculator.get_scores(candidate, reference, avg=True)['rouge-2']['f']

    # Rouge-L
    rouge_l = rouge_calculator.get_scores(candidate, reference, avg=True)['rouge-l']['f']

    print("Rouge-1:", rouge_1 * 100)
    print("Rouge-2:", rouge_2 * 100)
    print("Rouge-L:", rouge_l * 100)
    return rouge_1, rouge_2, rouge_l