Skip to content

chijames/structured_dialogue_discourse_parsing

Repository files navigation

Structured Dialogue Discourse Parsing

Pytorch implementation of the paper Structured Dialogue Discourse Parsing.

Installation

Run the following commands to recreate the environment we used for the development of this project:

conda create -n sddp python=3.9.6 -y
conda activate sddp
conda install pytorch==1.12.1 cudatoolkit=11.3 -c pytorch -y
pip install -r requirements.txt
pip install git+https://github.com/timvieira/arsenal.git@b7a5f93b04533236bbe297b4079256ea15812b88
pip install -e git+https://github.com/rycolab/spanningtrees.git@68c68e6f37d0951a524b901d9e84c5c1c498420e#egg=spanningtrees

Data Preparation

Please pass --link_only to all the shell scripts if you want to train and test using the link_only option. The default is link and relation.

cd data/
bash parse.sh (--link_only)
cd ../

Train/save on one domain

bash train_stac.sh (--link_only)
bash train_molweni.sh (--link_only)

Test on one domain

bash test_stac.sh (--link_only)
bash test_molweni.sh (--link_only)

Train on one domain and test/save on another domain

bash train_stac_on_molweni.sh (--link_only)
bash train_molweni_on_stac.sh (--link_only)

Test on another domain

bash test_stac_on_molweni.sh (--link_only)
bash test_molweni_on_stac.sh (--link_only)

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published