Update README.md
Browse files
README.md
CHANGED
|
@@ -33,16 +33,6 @@ The model was fine-tuned using the following datasets:
|
|
| 33 |
## Training Procedure
|
| 34 |
- The model was fine-tuned using the Hugging Face Transformer library. The base model, [gemma-7b-it](https://huggingface.co/google/gemma-7b-it), was further trained on the combined dataset of LeetCode user solutions and YouTube video captions(CoT Summary). This fine-tuning process was designed to enhance the model's understanding of coding concepts and problem-solving strategies, and improve its ability to generate relevant code snippets and explanations.
|
| 35 |
- The model was trained using the QLoRA technique with 4-bit quantization on the dataset.
|
| 36 |
-
|
| 37 |
-
## Bias and Limitations
|
| 38 |
-
- The model's knowledge is primarily based on the LeetCode user solutions and YouTube video captions(CoT Summary) used for fine-tuning. It may have limitations in handling coding problems or concepts that are not well-represented in the training data.
|
| 39 |
-
- The model's responses are generated based on patterns and information learned from the training data. It may sometimes produce incorrect or suboptimal solutions. Users should always review and verify the generated code before using it in practice.
|
| 40 |
-
- The model may exhibit biases present in the training data, such as favoring certain programming styles, algorithms, or approaches. It is important to consider alternative solutions and best practices when using the model's outputs.
|
| 41 |
-
|
| 42 |
-
## Ethical Considerations
|
| 43 |
-
- The model should be used as a supportive tool for learning and problem-solving, not as a substitute for human expertise and critical thinking.
|
| 44 |
-
- Users should be aware that the model's responses are generated based on patterns in the training data and may not always be accurate, complete, or up to date.
|
| 45 |
-
- The model should not be relied upon for making critical decisions or solving real-world problems without thorough validation and testing.
|
| 46 |
|
| 47 |
## Usage
|
| 48 |
To use the CodeMind model, you can access it through the Hugging Face model hub or by integrating it into your own applications using the provided API. Provide a coding problem or a question related to programming concepts, and the model will generate relevant explanations, code snippets, or guidance based on its training.
|
|
@@ -95,4 +85,14 @@ def get_completion(query: str, model, tokenizer) -> str:
|
|
| 95 |
|
| 96 |
result = get_completion(query="Tell me how to solve the Leetcode Two Sum problem", model=model, tokenizer=tokenizer)
|
| 97 |
print(result)
|
| 98 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
## Training Procedure
|
| 34 |
- The model was fine-tuned using the Hugging Face Transformer library. The base model, [gemma-7b-it](https://huggingface.co/google/gemma-7b-it), was further trained on the combined dataset of LeetCode user solutions and YouTube video captions(CoT Summary). This fine-tuning process was designed to enhance the model's understanding of coding concepts and problem-solving strategies, and improve its ability to generate relevant code snippets and explanations.
|
| 35 |
- The model was trained using the QLoRA technique with 4-bit quantization on the dataset.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
|
| 37 |
## Usage
|
| 38 |
To use the CodeMind model, you can access it through the Hugging Face model hub or by integrating it into your own applications using the provided API. Provide a coding problem or a question related to programming concepts, and the model will generate relevant explanations, code snippets, or guidance based on its training.
|
|
|
|
| 85 |
|
| 86 |
result = get_completion(query="Tell me how to solve the Leetcode Two Sum problem", model=model, tokenizer=tokenizer)
|
| 87 |
print(result)
|
| 88 |
+
```
|
| 89 |
+
|
| 90 |
+
## Bias and Limitations
|
| 91 |
+
- The model's knowledge is primarily based on the LeetCode user solutions and YouTube video captions(CoT Summary) used for fine-tuning. It may have limitations in handling coding problems or concepts that are not well-represented in the training data.
|
| 92 |
+
- The model's responses are generated based on patterns and information learned from the training data. It may sometimes produce incorrect or suboptimal solutions. Users should always review and verify the generated code before using it in practice.
|
| 93 |
+
- The model may exhibit biases present in the training data, such as favoring certain programming styles, algorithms, or approaches. It is important to consider alternative solutions and best practices when using the model's outputs.
|
| 94 |
+
|
| 95 |
+
## Ethical Considerations
|
| 96 |
+
- The model should be used as a supportive tool for learning and problem-solving, not as a substitute for human expertise and critical thinking.
|
| 97 |
+
- Users should be aware that the model's responses are generated based on patterns in the training data and may not always be accurate, complete, or up to date.
|
| 98 |
+
- The model should not be relied upon for making critical decisions or solving real-world problems without thorough validation and testing.
|