limohanlmh / Timestep-Aware-SentenceEmbedding-and-AcmeCoverage

Timestep-aware-SentenceEmbedding-and-AcmeCoverage (TSE_AC)

TSE_AC

Code of our paper "A Timestep aware Sentence Embedding and Acme Coverage for Brief but Informative Title Generation (NAACL-2022 findings)

Setup

To run the code, you need the following dependencies:

• Pytorch 1.4.0
• Spacy 2.3.2

Run

data processing: change the data path to yourself and run python data_process.py train the model: run python train_dynMSG.py and modify the configs you needed