hongwang600 / Summarization

Code for our ACL 2019 paper:

Self-Supervised Learning for Contextualized Extractive Summarization

Paper link: https://arxiv.org/abs/1906.04466

Prepare data

Download data

Requirement

Install pyrouge

pretrain the model

sh run_pretrain.sh

fine-tune the model

sh run_summarization.sh

evaluate the model

copy the model under 'results/models' to 'results/model_to_evaluate.pt'

sh run_eval.sh