This is an unoffical repo adapted from Repo. Followings are the differences:
| Desc | Official | This |
|---|---|---|
| Python Version | Python 2.7 | Python 2.7 - 3.7 |
| Tensorflow Version | 1.4 | 1.14 |
| CUDA Version | CUDA8.0 | CUDA10.0 |
| Code Run | Some issues | Can run |
| Platform | Linux | Support both Linux & Windows |
Dataset for ACL 2019 paper "Improving Multi-turn Dialogue Modelling with Utterance ReWriter "
After another two months of human labeling, we release a much more better quality dataset(only positive samples) than the original one we used in our paper for better research. Hope you can get a better result.
The positive dataset, 20000 dialogs. Each line in corpus.txt consists of four utterances of dialog (two context utterances, current utterance), and the rewritten uterance. Each line is tab-delimited (one tab) with the following format:
<A: context_1>\t<B: context_2>\t<A: current>\t<A: A: rewritten current>This code is based on the Pointer-Generator code.
Requirements
To run the souce codes, some external packages are required
- python 2.7
- Tensorflow 1.4
vocab file:
<word>\t<count>You may want to run a concurrent evaluation job, that runs your model on the validation set and logs the loss. To do this, run: To train your model, run:
sh train.sh
sh val.sh
To run beam search decoding, first set restore_best_model=1 to restore the best model.
sh train.sh
sh test.sh
Why can't you release the Transformer model? Due to the company legal policy reasons, we cannot realease the Transformer code which has been used in online environment. However, feel free to email us to discuss training and model details.
@article{su2019improving,
title={Improving Multi-turn Dialogue Modelling with Utterance ReWriter},
author={Su, Hui and Shen, Xiaoyu and Zhang, Rongzhi and Sun, Fei and Hu, Pengwei and Niu, Cheng and Zhou, Jie},
journal={ACL},
year={2019}
}