chibao24 commited on
Commit
27998a1
1 Parent(s): 7900a72

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -49,6 +49,8 @@ I got the idea from this [LLM classifier](https://github.com/lamini-ai/llm-class
49
  The model utilizes Few-Shot Learning techniques through SetFit, requiring only 8 examples per class. It can be trained in less than 1 minute on an RTX 3060 graphics card.
50
  This method provides an efficient solution for developing lightweight models suitable for real-world applications.
51
 
 
 
52
  This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) as the Sentence Transformer embedding model. A [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance is used for classification.
53
 
54
  The model has been trained using an efficient few-shot learning technique that involves:
 
49
  The model utilizes Few-Shot Learning techniques through SetFit, requiring only 8 examples per class. It can be trained in less than 1 minute on an RTX 3060 graphics card.
50
  This method provides an efficient solution for developing lightweight models suitable for real-world applications.
51
 
52
+ The source code can be found in my repo [mrzaizai2k/LLM-with-RAG](https://github.com/mrzaizai2k/LLM-with-RAG)
53
+
54
  This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) as the Sentence Transformer embedding model. A [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance is used for classification.
55
 
56
  The model has been trained using an efficient few-shot learning technique that involves: