Skip to content
/ nl2sql Public

Natural language query transformation to sql

Notifications You must be signed in to change notification settings

pmsgd/nl2sql

Repository files navigation

Installation

Install pip requirements:

pip3 install -r requirements.txt

Install spacy en model:

python3 -m spacy download en

Data preprocessing

Run preprocessing python script for all datasets:

mkdir inputsets outputsets models

python3 preprocess.py --queries ../WikiSQL/data/dev.jsonl --tables ../WikiSQL/data/dev.tables.jsonl --output inputsets/dev
python3 preprocess.py --queries ../WikiSQL/data/train.jsonl --tables ../WikiSQL/data/train.tables.jsonl --output inputsets/train
python3 preprocess.py --queries ../WikiSQL/data/test.jsonl --tables ../WikiSQL/data/test.tables.jsonl --output inputsets/test

Training

Run training on local machine (with --no_cuda if CUDA enabled GPU is missing):

python3 train.py --train_data inputsets/train --validation_data inputsets/test --save_model models/model

Translation

Run translation on local machine (with --no_cuda if CUDA enabled GPU is missing):

python3 translate.py --src inputsets/dev.en --model models/model

About

Natural language query transformation to sql

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •  

Languages