File size: 1,480 Bytes
6d446a9 5a93818 6d446a9 5a93818 6d446a9 a3cc44d 6d446a9 09ef8f6 5a93818 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
---
license: mit
datasets:
- b-mc2/sql-create-context
- gretelai/synthetic_text_to_sql
language:
- en
---
# Fine-Tuned Google T5 Model for Text to SQL Translation
A fine-tuned version of the Google T5 model, trained for the task of translating natural language queries into SQL statements.
## Model Details
- **Architecture**: Google T5 Base (Text-to-Text Transfer Transformer)
- **Task**: Text to SQL Translation
- **Fine-Tuning Datasets**:
- [sql-create-context Dataset](https://huggingface.co/datasets/b-mc2/sql-create-context)
- [Synthetic-Text-To-SQL Dataset](https://huggingface.co/datasets/gretelai/synthetic-text-to-sql)
## Ongoing Work
Currently working to implement PICARD (Parsing Incrementally for Constrained Auto-Regressive Decoding from Language Models) to improve the results of this model. More details can be found in the original [PICARD paper](https://arxiv.org/abs/2109.05093).
## Results
Results are currently being evaluated and will be posted here soon.
## Files
- `optimizer.pt`: State of the optimizer.
- `training_args.bin`: Training arguments and hyperparameters.
- `tokenizer.json`: Tokenizer vocabulary and settings.
- `spiece.model`: SentencePiece model file.
- `special_tokens_map.json`: Special tokens mapping.
- `tokenizer_config.json`: Tokenizer configuration settings.
- `model.safetensors`: Trained model weights.
- `generation_config.json`: Configuration for text generation.
- `config.json`: Model architecture configuration. |