ROUGE安装配置

ROUGE是做文本生成(eg文本摘要)的一个评价指标。
ROUGE的安装真的是一个巨坑,找了各种教程都不行。。实在搞不定只能按最复杂的教程搞了,最后居然成功了。。

  1. 首先,按照以下的方案配置:
    Ubuntu安装配置ROUGE
  2. pip install pyrouge(Python环境下调用rouge)
  3. 修改路径为ROUGE1.5.5的路径:pyrouge_set_rouge_path yourPath/ROUGE/RELEASE-1.5.5

但是!!!最后使用Rouge155()居然没有计算出来,气死我了,有种白忙活的感觉。。

仔细想一想,我使用的只是rouge-1, rouge-2, rouge-l,熟悉rouge的童鞋都知道,这只是一个简单的n-gram的召回,不可能这么复杂啊。。

后来,又查了一下资料,终于发现了一个简单的调用方法。
直接 pip install rouge
然后按照以下步骤使用即可:

from rouge import Rouge

a = ["i am a student from xx school"]  # 预测摘要 (可以是列表也可以是句子)
b = ["i am a student from school on china"] #真实摘要

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"])

注意,如果按照前面的配置流程配置过,那需要把vim ~/.profile 进入文件里,把下面这句话注释掉,不然rouge_score结果为空字典。

#export ROUGE_EVAL_HOME="$ROUGE_EVAL_HOME:/usr/local/RELEASE-1.5.5/data"

如果输入是中文,则要先分词,一句话中的每个词用空格隔开,再使用rouge评估。
参考:pyrouge和rouge,文本摘要评测方法库

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值