TarrySingh / TREQS

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

TREQS

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:

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

Ping Wang Tian Shi, Chandan K. Reddy

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

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