USC-Applied-NLP-Group
/

SQL-Generation

Model card Files Files and versions Metrics Training metrics Community

Ctrl+K

Ctrl+K

4 contributors

History: 61 commits

licesma's picture

Add a pre rag prompt

e029c99 10 days ago

__pycache__
updated rag evaluation function to match that from baseline testing 13 days ago
deepseek-coder-1.3b-instruct
Add column evaluator 15 days ago
fine-tuned-model-16
finished finetuning with 16 rank lora 13 days ago
fine-tuned-model-8-diff
trained 8 rank LoRA finetuned model, saving train and validation datasets to disk 16 days ago
fine-tuned-model-8
first cut at fine-tuning with masked training, working but very very slow for inference 17 days ago
nba-data
added nba kaggle sqlite database and testing python notebook 30 days ago
src
Add a pre rag prompt 10 days ago
train-data
Upload data_augmentation.tsv file (still need to go through queries and check if they are valid) 14 days ago
train.hf
trained 8 rank LoRA finetuned model, saving train and validation datasets to disk 16 days ago
val-16.hf
finished finetuning with 16 rank lora 13 days ago
val.hf
trained 8 rank LoRA finetuned model, saving train and validation datasets to disk 16 days ago
.gitattributes

1.56 kB

added nba kaggle sqlite database and testing python notebook 30 days ago
README.md

5.11 kB

Test 23 days ago
augment_data.ipynb

13.6 kB

first cut at adding data augmentation python notebook 18 days ago
database_setup.ipynb

1.81 kB

added nba kaggle sqlite database and testing python notebook 30 days ago
demo.py

5.42 kB

Renamed demo script and added initial pre-trained test python notebook 23 days ago
demo_rag.py

12 kB

Move paht to a variable 15 days ago
finetune_model.ipynb

214 kB

fix cuda 13 days ago
rag_metadata.py

2.3 kB

Create rag_metadata.py 15 days ago
test_finetuned.ipynb

31 kB

switch fine-tuned evaluation script to use 8 rank model 16 days ago
test_pretrained.ipynb

14.5 kB

Updated compare_result function to allow passing the cursor, also re-ran test_pretrained and test_rag with updated loss function 12 days ago
test_rag.ipynb

23.4 kB

Added testing on small database to test_rag notebook, removed excess print statements 12 days ago
test_rag.py

19.4 kB

Updating compare result function in src directory, fixing small issues in test_rag prompt, adding test_rag notebook 13 days ago