USC-Applied-NLP-Group
/

SQL-Generation

Model card Files Files and versions Metrics Training metrics Community

Ctrl+K

Ctrl+K

4 contributors

History: 64 commits

licesma's picture

Add second RAG notebook

1860a94 10 days ago

__pycache__
updated rag evaluation function to match that from baseline testing 14 days ago
deepseek-coder-1.3b-instruct
Add column evaluator 16 days ago
fine-tuned-model-16
finished finetuning with 16 rank lora 14 days ago
fine-tuned-model-8-diff
trained 8 rank LoRA finetuned model, saving train and validation datasets to disk 17 days ago
fine-tuned-model-8
first cut at fine-tuning with masked training, working but very very slow for inference 18 days ago
nba-data
added nba kaggle sqlite database and testing python notebook about 1 month ago
src
Add second RAG notebook 10 days ago
train-data
Add second RAG notebook 10 days ago
train.hf
trained 8 rank LoRA finetuned model, saving train and validation datasets to disk 17 days ago
val-16.hf
finished finetuning with 16 rank lora 14 days ago
val.hf
trained 8 rank LoRA finetuned model, saving train and validation datasets to disk 17 days ago
.gitattributes

1.56 kB

added nba kaggle sqlite database and testing python notebook about 1 month ago
README.md

5.11 kB

Test 23 days ago
augment_data.ipynb

13.6 kB

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

1.81 kB

added nba kaggle sqlite database and testing python notebook about 1 month ago
demo.py

5.42 kB

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

12 kB

Move paht to a variable 16 days ago
finetune_model.ipynb

214 kB

fix cuda 14 days ago
rag_helper.ipynb

9.38 kB

Add changes to save new rag dadataframe 11 days ago
rag_metadata.py

2.3 kB

Create rag_metadata.py 16 days ago
test_finetuned.ipynb

31 kB

switch fine-tuned evaluation script to use 8 rank model 17 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 13 days ago
test_rag.ipynb

23.4 kB

Added testing on small database to test_rag notebook, removed excess print statements 13 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 14 days ago
test_rag_2.ipynb

16.5 kB

Add second RAG notebook 10 days ago