qbuer/unisar_text2sql

"UniSAr: A Unified Structure-Aware Autoregressive Language Model for Text-to-SQL"

Introduction

This paper introduces UniSAr.

Dataset and Model

Spider -> ./data/spider

Fine-tuned BART model -> ./models/spider_sl

(Please download this model by git-lfs to avoid the issue.)

git lfs install
git clone https://huggingface.co/dreamerdeo/mark-bart

Main dependencies

• Python version >= 3.6
• PyTorch version >= 1.5.0
• pip install -r requirements.txt
• fairseq is going though changing without backward compatibility. Install fairseq from source and use this commit for reproducibilty. See here for the current PR that should fix fairseq/master.

(Since the constrained decoding is implemented base on GENRE, please install the specific fairseq version as README suggested.)

git clone [email protected]:nicola-decao/fairseq.git
cd fairseq
pip install --editable ./

Evaluation Pipeline

Step 1: Preprocess via adding schema-linking and value-linking tag.

python step1_schema_linking.py

Step 2: Building the input and output for BART.

python step2_serialization.py

Step 3: Evaluation Script with/without constrained decoding.

python step3_evaluate.py --constrain

Results

Prediction: 69.34

Prediction with Constrain Decoding: 70.02

Interactive

python interactive.py --logdir ./models/spider-sl --db_id student_1 --db-path ./data/spider/database --schema-path ./data/spider/tables.json

Reference Code

https://github.com/ryanzhumich/editsql

https://github.com/benbogin/spider-schema-gnn-global

https://github.com/ElementAI/duorat

https://github.com/facebookresearch/GENRE