文件名称:MatchSum:ACL 2020纸代码
文件大小:70KB
文件格式:ZIP
更新时间:2024-04-21 07:29:06
Python
匹配总和 ACL 2020论文代码: 依存关系 Python 3.7 1.4.0 0.5.0 0.1.3 在运行我们的代码之前,您应该在metrics.py第20行中填写ROUGE路径。 1.0.0 在验证阶段使用。 2.5.1 所有代码仅支持在Linux上运行。 数据 我们已经处理了CNN / DailyMail数据集,您可以通过下载它,解压缩并将其移动到./data 。 它包含数据集的两个版本(BERT / RoBERTa),总共六个文件。 此外,我们还发布了其他五个已处理的数据集(WikiHow,PubMed,XSum,MultiNews,Reddit),您可以在找到。 火车 我们使用八个Tesla-V100-16G GPU来训练我们的模型,训练时间约为30小时。 如果你没有足够的显存,可以减少或batch_size时在candidate_num train_mat
【文件预览】:
MatchSum-master
----utils.py(920B)
----metrics.py(8KB)
----dataloader.py(3KB)
----train_matching.py(6KB)
----preprocess()
--------test_cnndm.jsonl(369KB)
--------get_candidate.py(7KB)
----model.py(2KB)
----callback.py(1KB)
----README.md(5KB)