snakeztc / simdial

Synthetic task-oriented dialog generator with controllable complexity. It is released by Tiancheng Zhao (Tony) from Dialog Research Center, LTI, CMU

CMU SimDial: Synthetic Task-oriented Dialog Generator with Controllable Complexity

This is the dialog data used by our SIGDIAL 2018 paper: Zero-Shot Dialog Generation with Cross-Domain Latent Actions. See paper for details. The source code and data used for the paper can be found at here.

Prerequisites

• Python 2.7
• Numpy
• NLTK
• progressbar

Usage

Run the following code to generate dialog data for multiple domains that are defined in the multiple_domains.py script.

python multiple_domains.py

The data will be saved into two folders

test/ for testing data 
train/ for training data

References

If you use any source codes or datasets included in this toolkit in your work, please cite the following paper. The bibtex are listed below:

@article{zhao2018zero,
  title={Zero-Shot Dialog Generation with Cross-Domain Latent Actions},
  author={Zhao, Tiancheng and Eskenazi, Maxine},
  journal={arXiv preprint arXiv:1805.04803},
  year={2018}
}

simdial's Issues

Error while running multiple_domains.py

Traceback (most recent call last):
  File "mtrand.pyx", line 1115, in mtrand.RandomState.choice
TypeError: 'dict_keys' object cannot be interpreted as an integer

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "multiple_domains.py", line 311, in <module>
    gen_bot.gen_corpus("test", rest_spec, complexity.CleanSpec, test_size)
  File "C:\Users\Mayank\Desktop\GithubProjects\SimDial\simdial\generator.py", line 146, in gen_corpus
    corpus = self.gen(domain, complex, num_sess=size)
  File "C:\Users\Mayank\Desktop\GithubProjects\SimDial\simdial\generator.py", line 108, in gen
    usr = User(domain, complexity)
  File "C:\Users\Mayank\Desktop\GithubProjects\SimDial\simdial\agent\user.py", line 76, in __init__
    self.goal_cnt = np.random.choice(complexity.multi_goals.keys(), p=complexity.multi_goals.values())
  File "mtrand.pyx", line 1117, in mtrand.RandomState.choice
ValueError: 'a' must be 1-dimensional or an integer

what's the version of progressbar used?