A Translate-Edit Model for Question to SQL Query Generation
- 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
- Another version of TREQS can be found in LeafNATS.
The MIMICSQL dataset will be released soon.
-
Training:
python main.py -
Validate:
python main.py --task validate -
Test:
python main.py --task test
@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}
}