Zero-shot Text-to-SQL Learning with Auxiliary Task

Code for Zero-shot Text-to-SQL Learning with Auxiliary Task

Usage

Conda Environments

Please use Python 3.6 and Pytorch 1.3. Other Python dependency is in requirement.txt. Install Python dependency with:

	pip install -r requirements.txt

Download Data

Data can be found from google drive. Please download them and extract them into root path.

Generate our respilted WikiSQL data

	cd data_model/wikisql
	python make_zs.py
	python make_fs.py

Run the model on original WikiSQL and our split

	cd zero-shot-text-to-SQL
	./run.sh  GPU_ID

Acknowledgement

• This implementation is based on coarse2fine.
• The preprocessing and evaluation code used for WikiSQL is from salesforce/WikiSQL.