witkeyshare/treqs

A Translate-Edit Model for Question to SQL Query Generation in Healthcare

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:

A Translate-Edit Model for Natural Language Question to SQL Query Generation on Multi-relational Healthcare Data

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}
}