Skip to main content

Convert natural language questions to SQL query

Project description

Text2SQL

How many times have you pulled your hair apart writing a SQL query, now use natural language to convert to appropriate SQL and save your precious hair.

Though this can be used as a standalone package, I highly recommend that you use streamlit to play with the model interactively, to run it interactively

streamlit run t2s.py

Installation

To install this repo clone it then run setup.py

git clone https://github.com/yashbonde/text2sql.git
cd text2sql/
pip3 install -e . #text2sql

Parsing

New method of parsing to convert each DB to a graph network, red denotes foreign keys.

Trainer File

Simple trainer file.

Datasets

Using CoSQL, Spider, Sparc datasets, credit to the authors. There are a couple of things to note, we have in total 178 tables, but only 166 tables in training date and dev set has 20 tables.

License

Following are the licensing things:

  1. I am releasing my part of code under MIT License
  2. transformers code is under Apache License, Version 2.0

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

text2sql-0.1.1.tar.gz (2.9 kB view hashes)

Uploaded source

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page