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