This is the code for our paper Global Encoding for Abstractive Summarization, https://arxiv.org/abs/1805.03989
- Ubuntu 16.0.4
- Python 3.5
- Pytorch 0.3.1
- pyrouge
In order to use pyrouge, set rouge path with the line below:
pyrouge_set_rouge_path script/RELEASE-1.5.5
python3 preprocess.py -load_data path_to_data -save_data path_to_store_data
Remember to put the data into a folder and name them train.src, train.tgt, valid.src, valid.tgt, test.src and test.tgt, and make a new folder inside called data
python3 train.py -log log_name -config config_yaml -gpus id
python3 train.py -log log_name -config config_yaml -gpus id -restore checkpoint -mode eval
If you use this code for your research, please cite the paper this code is based on: Global Encoding for Abstractive Summarization:.
@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}
}