codemayq / coarse2fine

Coarse-to-Fine Decoding for Neural Semantic Parsing

Setup

Requirements

• Python 3.5
• PyTorch 0.2.0.post3 (GPU)

Install Python dependency

pip install -r requirements.txt

Download data and pretrained models

Download the zip file from Google Drive, and copy it to the root folder.

unzip acl18coarse2fine_data_model.zip

Usage

Run pretrained models

./pretrain.sh [geoqueries|atis|django|wikisql] GPU_ID

Run experiments

./run.sh [geoqueries|atis|django|wikisql] GPU_ID

Acknowledgments

• The implementation is based on OpenNMT/OpenNMT-py.
• The preprocessing and evaluation code used for WikiSQL is from salesforce/WikiSQL.