Speaker-Aware Discourse Parsing on Multi-Party Dialogues

Implementation of the paper Speaker-Aware Discourse Parsing on Multi-Party Dialogues, COLING'22

Requirements

PyTorch 1.9.1+cu111

Python 3.9

Transformers 4.21.1

Data

STAC

Molweni

PLM

For Molweni: SSP-BERT

For STAC: SSP-BERT

Training

python -u driver/Train.py --config_file molweni.cfg

Testing

We provide the trained models (molweni_ddp_model, stac_ddp_model) for testing.

python -u driver/Test.py --config_file ddp_model/config.cfg