DoctorChaos/text-to-SQL-clause-smbop

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

Model Details

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