A Translate-Edit Model for Question to SQL Query Generation
Our new implementation is available at https://github.com/wangpinggl/TREQS. Please check it.
- This repository is a pytorch implementation of the Translate-Edit Model for Question to SQL Query Generation proposed in the following paper:
Ping Wang Tian Shi, Chandan K. Reddy
The MIMICSQL dataset will be released soon.
-
Training:
python main.py -
Validate:
python main.py --task validate -
Test:
python main.py --task test
If you use the code, please kindly consider citing our paper:
@article{wang2019treqs,
title={A Translate-Edit Model for Natural Language Question to SQL Query Generation on Multi-relational Healthcare Data},
author={Wang, Ping and Shi, Tian and Reddy, Chandan K.},
journal={arXiv preprint arXiv:1908.01839},
year={2019}
}