1、pip install rouge
2、代码:
from rouge import Rouge
import os
rouge = Rouge()
cand_path=r"F:\checkpoint\result\logcandidate"
ref_path=r"F:\checkpoint\result\logreference"
for i in range(11):
cand_name="cand.%d.txt"%i
ref_name="ref.%d.txt"%i
cand_dir=os.path.join(cand_path,cand_name)
ref_dir=os.path.join(ref_path,ref_name)
cand=open(cand_dir).readlines()
ref=open(ref_dir).readlines()
rouge_score = rouge.get_scores(cand,ref)
print(rouge_score[0]["rouge-1"])
print(rouge_score[0]["rouge-2"])
print(rouge_score[0]["rouge-l"])
3、修改代码里路径之类的
PS:rouge安装太恶心了,windows安装Linux安装都尝试过了,永远缺东西。用了ActivePerl 、strawberry Perl,还要安装各种依赖库,XML::DOM,XML::Parse之类啥的永远都没安装好。后来忘记在哪看到这个方法,记录一下,以免下次再需要用rouge的时候又被搞心态