Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: Knowledge Graph Retriever trainer using PEFT #39

Merged
merged 7 commits into from
Aug 25, 2024

Conversation

debrupf2946
Copy link
Collaborator

This pull request aims to add a few modules and experiments with regard to Knowledge Graph retrieval

  • p-tuning retriever on custom data, used for retrieval of KG
  • QLora trained retriever on custom data, used for retrieval of KG

@kmehant
Copy link
Member

kmehant commented Aug 19, 2024

Going in the right direction.

@kmehant
Copy link
Member

kmehant commented Aug 19, 2024

Please add a detailed readme @debrupf2946

@debrupf2946 debrupf2946 marked this pull request as ready for review August 24, 2024 16:32
Copy link
Member

@kmehant kmehant left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Great Job! @debrupf2946

@kmehant kmehant merged commit 86252ac into c2siorg:main Aug 25, 2024
1 check passed
AsH1605 pushed a commit to AsH1605/Project-Explainer that referenced this pull request Nov 26, 2024
* started adding training scripts initial commit

* modified training scripts and added QLoRA training scripts

* added arg-parse functionality to run training scripts

* formatted the scripts using black

* added doc strings for the training files

* added a markdown file domenstrating my work on property graph and relick for relation extraction which has further scope of bieng intigrated into KG builder

* updated README.MD file for training scripts
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants