- 安装Rouge:
pip install rouge
- 计算生成文本与 参考文本的Rouge值
# coding:utf8
from rouge import Rouge
a = ["i am a student from china"] # 预测摘要 （可以是列表也可以是句子）
b = ["i am student from school on japan"] #真实摘要
'''
f:F1值 p：查准率 R：召回率
'''
rouge = Rouge()
rouge_score = rouge.get_scores(a, b)
print(rouge_score[0]["rouge-1"])
print(rouge_score[0]["rouge-2"])
print(rouge_score[0]["rouge-l"])
- 结果如下：
{'p': 1.0, 'f': 0.7272727226446282, 'r': 0.5714285714285714}
{'p': 1.0, 'f': 0.6666666622222223, 'r': 0.5}
{'p': 1.0, 'f': 0.6388206388206618, 'r': 0.5714285714285714}
- 参考：https://blog.csdn.net/u013521274/article/details/89460322
