欢迎访问宙启技术站
智能推送

用pyrouge比较两个自动摘要系统生成的摘要的内容相似度。

发布时间:2023-12-26 08:42:04

使用pyrouge比较两个自动摘要系统生成的摘要的内容相似度可以通过以下步骤实现。

首先,确保你已经安装了pyrouge库。

!pip install pyrouge

接下来,准备两个摘要文件,每个文件包含一个或多个自动生成的摘要句子。例如,我们有两个摘要文件"summary1.txt"和"summary2.txt",每个文件有一个句子。

summary1.txt内容如下:

自动摘要系统生成的句子。

summary2.txt内容如下:

另一个自动摘要系统生成的句子。

然后,编写如下代码来比较两个摘要文件的内容相似度。

from pyrouge import Rouge155

# 创建Rouge155对象
rouge = Rouge155()

# 设置系统生成的摘要文件路径
rouge.system_dir = 'system_summaries'
rouge.system_filename_pattern = 'summary.(\d+).txt'

# 设置参考摘要文件路径
rouge.reference_dir = 'reference_summaries'
rouge.reference_filename_pattern = 'summary.[A-Z].txt'

# 计算摘要的评估得分
output = rouge.convert_and_evaluate()
print(output)

# 读取评估结果文件
output_dict = rouge.output_to_dict(output)
print(output_dict)

上述代码中,我们首先创建了Rouge155对象,然后设置了系统生成的摘要文件的路径和参考摘要文件的路径。system_filename_pattern和reference_filename_pattern用于指定文件命名的模式。接下来,我们调用convert_and_evaluate方法来计算摘要的评估得分,并将得分的输出保存在output变量中。最后,我们使用output_to_dict方法将评估结果转换为字典类型,并打印输出。

运行上述代码后,将得到两个摘要系统生成摘要的内容相似度得分,包括ROUGE-N和ROUGE-L等指标的得分。具体的评估结果可以在output_dict中查看。