T5-small finedtuned to generate txtai SQL
T5 small fine-tuned to generate txtai SQL. This model takes Bash like commands and builds txtai-compatible SQL statements.
find -name "feel good story" -mtime -1
find -name "show me sports stories" -mtime -1 -team \"Red Sox\"
find -name "breaking news" -summary
find -name "breaking news" -translate fr
Custom query syntax
This model is an example of creating a custom query syntax that can be translated into SQL txtai can understand. Any query syntax can be created. This one supports Bash-like commands but a similar strategy can be deployed to support other languages. Natural language can be translated to functions, query clauses, column selection and more.
See t5-small-txtsql for a model that translates natural language statements into txtai SQL.
Model training
This model was trained using scripts that can be found here.
Steps to train:
python generate.py bashsql.csv
python train.py bashsql.csv t5-small-bashsql
- Downloads last month
- 11