Commit History

  1. add gpt4 generated queries on test

    1570fb6

  1. clean rag training

    e112bff
  2. Added fine-tuning python notebook for rag

    affe2f6
  3. add train rag file

    d83c5d0
  4. Notebook for augmented dataset

    4fa498e

  1. Fix bug

    1cb1219
  2. Second attempt of dynamic rag

    37aa3d3

  1. adding latest test dataset run

    c0c5b84
  2. Adding test set generation python notebook, and correspond train and test dataset tsv files

    99a8882
  3. Updated Augmentation Dataset with 250 verified queries

    Verified signature
    753e538

  1. Notebooks for GPT evaluation

    860fee5
  2. Fix typo in rag

    9a3c9fb
  3. Add second RAG notebook

    1860a94

  1. Add changes to save new rag dadataframe

    2ea9086
  2. Add a rag helper notebook

    59f20cf
  3. Add a pre rag prompt

    e029c99

  1. Added testing on small database to test_rag notebook, removed excess print statements

    57326c2
  2. Updated compare_result function to allow passing the cursor, also re-ran test_pretrained and test_rag with updated loss function

    23a14a5

  1. Tweaked compare_result to work with proper directory, updated rag python notebook prompt and vector stores to increase accuracy

    4022db3
  2. Updating compare result function in src directory, fixing small issues in test_rag prompt, adding test_rag notebook

    0405efb
  3. fix cuda

    ef32c53
  4. Fix colab bugs

    9687f25
  5. Prepare fine-tune for colab

    569052e
  6. updated rag evaluation function to match that from baseline testing

    2b3100e
  7. Update test_rag.py

    Verified signature
    ace09b0
  8. Merge branch 'higher_rank' into main

    5330d88
  9. finished finetuning with 16 rank lora

    3ad5864
  10. Fix bug on function evaluation

    cd9fee3

  1. Upload data_augmentation.tsv file (still need to go through queries and check if they are valid)

    Verified signature
    64e6526
  2. Modify path to relative path

    5af7feb
  3. Add relative path functionality for colab

    aa43cb9
  4. colab support part two

    c397e97
  5. Add support for colab

    bfad6ce
  6. Add a source folder

    c1d6d12

  1. Move paht to a variable

    30ff71b
  2. Create demo_rag.py

    Verified signature
    6be6272
  3. Create rag_metadata.py

    Verified signature
    6cb0a90
  4. Create test_rag.py

    Verified signature
    03182a1
  5. Add column evaluator

    2cfc97b
  6. added fine-tune testing on validation set with short queries only

    f94c0b8
  7. Adding validation set testing of newly fine-tuned model

    98e9ca0

  1. switch fine-tuned evaluation script to use 8 rank model

    da19664
  2. trained 8 rank LoRA finetuned model, saving train and validation datasets to disk

    c72582f

  1. first cut at fine-tuning with masked training, working but very very slow for inference

    a30a6cf
  2. fixed bug in test_finetuned notebook to read from proper directory

    271b076
  3. Storing 16 and 32 rank LoRA finetuned models

    48d9ca0
  4. Increasing LoRA rank to 32 and updating loss function

    c78bae0

  1. first cut at adding data augmentation python notebook

    b0335cf

  1. added python notebook for testing finetuned model

    62c7f8d

  1. First decent attempt, seems to generate valid queries after pre-training. Needs evaluation still

    fdaf162