
BertSum: 该代码对BERT模型进行微调,以提取摘要。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
BertSum 代码旨在对纸质文档进行Fine-tune,以实现Extractive Summarization 的目标! 值得注意的是,您可以参考我们提供的预训练模型,这些模型包含了CNN/Dailymail数据集的成果,日期为2019年3月25日。 具体指标包括:楷模的 ROUGE-1 得分40.9,ROUGE-2 得分18.02,以及ROUGE-L 得分37.17。 此外,与基于Transformer的基线模型相比,BERTSUM +分类器取得了43.23 的 ROUGE-1 分数、20.22 的 ROUGE-2 分数和39.60 的 ROUGE-L 分数。 同样,BERTSUM 与 Transformer 结合的模型表现出43.25 的 ROUGE-1 分数、20.24 的 ROUGE-2 分数和39.63 的 ROUGE-L 分数。 最后,BERTSUM 与 LSTM 模型的性能为43.22 的 ROUGE-1 分数、20.17 的 ROUGE-2 分数和39.59 的 ROUGE-L 分数。 该代码支持 Python 3.6 环境,并依赖于以下软件包:pytorch、pytorch_pretrained_bert 和 tensorboardX。 同时,部分代码片段借鉴自 ONMT 项目(...)。
全部评论 (0)
还没有任何评论哟~


