Skip to content
Code for SIGDial Paper: Structured Fusion Networks for Dialog https://arxiv.org/abs/1907.10016
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
db all code Aug 19, 2019
models all code Aug 19, 2019
utils all code Aug 19, 2019
README.md Create README.md Aug 19, 2019
data.zip all code Aug 19, 2019
delexicalize.py all code Aug 19, 2019
eval.py all code Aug 19, 2019
evaluate.py all code Aug 19, 2019
evaluate_model.py all code Aug 19, 2019
evaluate_model.pyc all code Aug 19, 2019
mapping.pair all code Aug 19, 2019
model.py all code Aug 19, 2019
pytorch.yml all code Aug 19, 2019
rl_eval.py all code Aug 19, 2019
rl_train.py all code Aug 19, 2019
temp.json all code Aug 19, 2019
test.py all code Aug 19, 2019
train.py all code Aug 19, 2019

README.md

Structured Fusion Networks

Structured Fusion Networks for Dialog https://arxiv.org/abs/1907.10016

Code written by Shikib Mehri and Tejas Srinivasan.

Pretrained Models

The main models trained in the paper have been provided and can be used as follows:

Sequence-to-Sequence with Attention:

python3 test.py --model_name models/attn --use_attn --test_only

Structured Fusion Networks:

python3 test.py --model_name models/sfn --structured_fusion --test_only

Structured Fusion Networks with Reinforcement Learning:

python3 test.py --model_name models/sfn_rl --structured_fusion --test_only

Training

Structured Fusion Networks can be trained with the following command:

python3 train.py --model_name 'models/sfn' --structured_fusion --use_cuda True --tune_params True

SFNs can be fine-tuned with RL as follows (you may need to edit the code a bit):

python3 rl_train.py --structured_fusion --model_name models/rl_sfn --lr 0.00001

Questions

Please feel free to send any questions/concerns to amehri@andrew.cmu.edu.

Citing

The following paper should be cited if you use this code.

@article{mehri2019structured,
  title={Structured Fusion Networks for Dialog},
  author={Mehri, Shikib and Srinivasan, Tejas and Eskenazi, Maxine},
  journal={arXiv preprint arXiv:1907.10016},
  year={2019}
}
You can’t perform that action at this time.