Downloads · 30 days
0
DoctorChaos/text-to-SQL-clause-smbop
text-to-SQL-clause-smbop is a machine learning model from DoctorChaos. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
We train a text-to-clause model for SQL generation. Given a natural language description as input, the model outputs the SQL clause (e.g., WHERE student.age 18) instead of the entire SQL query.
Downloads · 30 days
0
Access
Public
Updated Apr 20, 2024
Repo size
1.4 GB
Likes
1
Public
Click a slice to open those files.
.gz1.4 GB · 100%
From the Hugging Face model README
We train a text-to-clause model for SQL generation. Given a natural language description as input, the model outputs the SQL clause (e.g., WHERE student.age > 18) instead of the entire SQL query.
modelS.tar.gz includes our best checkpoint. Please refer below information for more details.
| Type | Examples |
|---|---|
| Application | Microsoft SQL Server, PostgreSQL, SQLite |
| Language | C++, Java, Python, Rust |
| Library | PyTorch, AllenNLP |
| License | MIT License |
| Decoding Architecture | Transformer + RAT-SQL + rule-based |
| Layers | 24 Transformer layers + 8 RAT-SQL layers |
| Tree representation | 8 heads & dimensionality 256 |
| Dataset | 83K synthetic dataset |
| Optimizer | Adam |
| Learning rate | 1.8e-4 |
| Dropout rate | 0.1 |
| Operating System | Linux, macOS, Windows |
Paper: https://arxiv.org/abs/2305.07372
Code: https://github.com/magic-YuanTian/STEPS
Dataset: https://drive.google.com/file/d/1f1fnJK2vGuRpaQOeMlBD10tQMDH3dR83/view?pli=1