transformers/examples/summarization/bart
2020-03-24 21:00:24 -04:00
..
__init__.py Summarization Examples: add Bart CNN Evaluation (#3082) 2020-03-03 15:29:59 -05:00
evaluate_cnn.py Clean special token init in modeling_....py (#3264) 2020-03-20 21:41:04 +01:00
README.md BART for summarization training with CNN/DM using pytorch-lightning 2020-03-24 21:00:24 -04:00
run_bart_sum.py BART for summarization training with CNN/DM using pytorch-lightning 2020-03-24 21:00:24 -04:00
run_train.sh BART for summarization training with CNN/DM using pytorch-lightning 2020-03-24 21:00:24 -04:00
test_bart_examples.py Summarization Examples: add Bart CNN Evaluation (#3082) 2020-03-03 15:29:59 -05:00
utils.py BART for summarization training with CNN/DM using pytorch-lightning 2020-03-24 21:00:24 -04:00

Get the CNN Data

To be able to reproduce the authors' results on the CNN/Daily Mail dataset you first need to download both CNN and Daily Mail datasets from Kyunghyun Cho's website (the links next to "Stories") in the same folder. Then uncompress the archives by running:

tar -xvf cnn_stories.tgz && tar -xvf dailymail_stories.tgz

this should make a directory called cnn_dm/ with files like test.source. To use your own data, copy that files format. Each article to be summarized is on its own line.

Usage

To create summaries for each article in dataset, run:

python evaluate_cnn.py <path_to_test.source> cnn_test_summaries.txt

the default batch size, 8, fits in 16GB GPU memory, but may need to be adjusted to fit your system.

Training

After downloading the CNN and Daily Mail datasets, preprocess the dataset:

git clone https://github.com/artmatsak/cnn-dailymail
cd cnn-dailymail && python make_datafiles.py ../cnn/stories/ ../dailymail/stories/

Run the training script: run_train.sh

Where is the code?

The core model is in src/transformers/modeling_bart.py. This directory only contains examples.

(WIP) Rouge Scores

Stanford CoreNLP Setup

ptb_tokenize () {
    cat $1 | java edu.stanford.nlp.process.PTBTokenizer -ioFileList -preserveLines > $2
}

sudo apt install openjdk-8-jre-headless
sudo apt-get install ant
wget http://nlp.stanford.edu/software/stanford-corenlp-full-2018-10-05.zip
unzip stanford-corenlp-full-2018-10-05.zip
cd stanford-corenlp-full-2018-10-05
export CLASSPATH=stanford-corenlp-3.9.2.jar:stanford-corenlp-3.9.2-models.jar

Then run ptb_tokenize on test.target and your generated hypotheses.

Rouge Setup

Install files2rouge following the instructions at here. I also needed to run sudo apt-get install libxml-parser-perl

from files2rouge import files2rouge
from files2rouge import settings
files2rouge.run(<path_to_tokenized_hypo>,
                <path_to_tokenized_target>,
               saveto='rouge_output.txt')