Dynamic Hybrid Relation Network for Cross-Domain Context-Dependent Semantic Parsing
R²SQL
The PyTorch implementation of paper Dynamic Hybrid Relation Network for Cross-Domain Context-Dependent Semantic Parsing. (AAAI 2021)
Requirements
The model is tested in python 3.6 with following requirements:
torch==1.0.0
transformers==2.10.0
sqlparse
pymysql
progressbar
nltk
numpy
six
spacy
All experiments on SParC and CoSQL datasets were run on NVIDIA V100 GPU with 32GB GPU memory.
- Tips: The 16GB GPU memory may appear out-of-memory error.
Setup
The SParC and CoSQL experiments in two different folders, you need to download different datasets from [SParC | CoSQL] to the {sparc|cosql}/data folder separately.
Another related data file could be download from EditSQL.
Then, download the database sqlite files from [here] as data/database.
Download Pretrained BERT model from [here] as model/bert/data/annotated_wikisql_and_PyTorch_bert_param/pytorch_model_uncased_L-12_H-768_A-12.bin.
Download Glove embeddings file (glove.840B.300d.txt) and change