Pytorch使用全局编码(Global-Encoding)生成文本摘要(北大开源)
Song •
1665 次浏览 •
0个评论 •
2019年03月14日
Global-Encoding
此项目主要是为了实现论文Global Encoding for Abstractive Summarization
一、要求
- Ubuntu 16.0.4
- Python 3.5
- Pytorch 0.4.1(更新)
- pyrouge
要使用
pyrouge
,请使用以下行设置rouge
路径:
pyrouge_set_rouge_path RELEASE-1.5.5/
似乎有些用户遇到了pyrouge
的问题,所以我更新了脚本,用户可以将目录RELEASE-1.5.5
放在你的主目录中并设置路由路径(或运行命令chmod 777 RELEASE-1.5.5
获得系统权限)。
二、预处理
python3 preprocess.py -load_data path_to_data -save_data path_to_store_data
记得将数据放入一个文件夹,并将它们命名为train.src
,train.tgt
,valid.src
,valid.tgt
,test.src
和test.tgt
,并在其中创建一个名为data
的新文件夹
三、训练
python3 train.py -log log_name -config config_yaml -gpus id
为超参数设置创建自己的yaml文件。
四、评估
python3 train.py -log log_name -config config_yaml -gpus id -restore checkpoint -mode eval
五、引文
如果您将此代码用于研究,请引用我们的论文:
@inproceedings{globalencoding,
title = {Global Encoding for Abstractive Summarization},
author = {Junyang Lin and Xu Sun and Shuming Ma and Qi Su},
booktitle = {{ACL} 2018},
year = {2018}
}
用户评论
当前暂无评价,快来发表您的观点吧...
更多相关好文
当前暂无更多相关好文推荐...
-
微信公众号文章/菜单添加小程序时路径如何获取? 2021-12-22
-
如何轻松获取微信小程序路径path? 2021-12-22
-
cannot import name 'CUDA_HOME' from 'mmcv.utils' 2021-12-05
-
vgg的loss一轮达到ln(1/n)阈值,如何解决 2021-11-21
-
如何下载使用utils库 2021-10-27
热门文章
-
微信公众号文章/菜单添加小程序时路径如何获取? 2021-12-22
-
如何轻松获取微信小程序路径path? 2021-12-22
-
python/MySQL分页查询方法与性能优化 2021-06-23
-
mitmproxy & python 忽略所有的https/ssl请求 2021-04-19
-
如何使用邮件/邮箱推广微信公众号/小程序? 2021-01-28
栏目最新文章
公告提示
- pytorch中文文档
- pytorch官方文档
提交评论