TREQS
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.
Dataset
The MIMICSQL dataset will be released soon.
Usuage
-
Training:python main.py -
Validate:python main.py --task validate -
Test:python main.py --task test
Citation
@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}
}