The code for "An Auto-Encoder Matching Model for Learning Utterance-Level Semantic Dependency in Dialogue Generation" (EMNLP 2018)
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
fetch initial commit Aug 20, 2018
model minor Aug 27, 2018
process initial commit Aug 20, 2018
.gitignore initial commit Aug 20, 2018
README.md update README Aug 27, 2018
config.yaml initial commit Aug 20, 2018
play.py update running mode to train Aug 27, 2018

README.md

Auto-Encoder Matching Model

The code for "An Auto-Encoder Matching Model for Learning Utterance-Level Semantic Dependency in Dialogue Generation"

Requirements

  • Python 3
  • Tensorflow >= 1.8
  • mlbootstrap == 0.02

Data Preparation

To use your own data, create a folder data/source/<dataset-name>/ and place the original data in the directory. Then write a parsing script (you can refer to daily.py) and update the config.yaml to include the new data path.

Training

python play.py

Evaluation

Change the last line in play.py to bootstrap.evaluate() and run python play.py

Hyperparameters

You can change the hyperparameters in config.yaml according to your needs.