TTSG

Env

Run the following command to create the required conda environment:

conda create -n TTSG python=3
conda activate TTSG
conda install pytorch cudatoolkit=11 -c pytorch

All dependencies can be installed via:

pip install -r requirements.txt

Datasets

20News Group

The 20News Group dataset can be accessed from here and the downloaded file is placed in the data directory.

BBC News

The BBC News dataset can be accessed from here and the downloaded file is placed in the data directory.

How to run script

Summary Generation

python BRIO/run_summary.py

In this section, we use the BRIO approach to generate summaries.

Document Summary Clustering

python k-cloest.py

Topic Sentence Generation

bash centrum/topic.sh

Here, we use Centrum method to generate topic sentences.

Acknowledgement

We appreciate BRIO, Centrum and many other related works for their open-source contributions.