Update README.md
Browse files
README.md
CHANGED
|
@@ -7,11 +7,20 @@ tags:
|
|
| 7 |
- knowledge extraction
|
| 8 |
- tiny
|
| 9 |
- small
|
|
|
|
| 10 |
---
|
|
|
|
|
|
|
| 11 |
A model that can **extract the knowledge points** from the given **C language code**.
|
| 12 |
|
| 13 |
The base model is [pythia-70m](https://huggingface.co/EleutherAI/pythia-70m). This model was fine-tuned with 10 epochs using [Q-Lora](https://github.com/artidoro/qlora) method on my own training set.
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
A usage example is as follows, first import the model and prepare the code:
|
| 16 |
|
| 17 |
```python
|
|
@@ -66,6 +75,8 @@ response = tokenizer.decode(tokens[0]).split('```')[-1].split('<')[0]
|
|
| 66 |
|
| 67 |
|
| 68 |
|
|
|
|
|
|
|
| 69 |
However, in practical use, in order to achieve more diverse representations, it's recommended to do multiple inferences. Don't worry, it's really small so the inferences don't take much time, as follows:
|
| 70 |
|
| 71 |
```python
|
|
@@ -95,5 +106,5 @@ print(ans_dict)
|
|
| 95 |
### 'Quick sort': 25,
|
| 96 |
### 'Recurrence': 2,
|
| 97 |
### 'Queue': 1
|
| 98 |
-
###}
|
| 99 |
```
|
|
|
|
| 7 |
- knowledge extraction
|
| 8 |
- tiny
|
| 9 |
- small
|
| 10 |
+
- C
|
| 11 |
---
|
| 12 |
+
## Model info
|
| 13 |
+
|
| 14 |
A model that can **extract the knowledge points** from the given **C language code**.
|
| 15 |
|
| 16 |
The base model is [pythia-70m](https://huggingface.co/EleutherAI/pythia-70m). This model was fine-tuned with 10 epochs using [Q-Lora](https://github.com/artidoro/qlora) method on my own training set.
|
| 17 |
|
| 18 |
+
|
| 19 |
+
|
| 20 |
+
## How to use
|
| 21 |
+
|
| 22 |
+
### quick start
|
| 23 |
+
|
| 24 |
A usage example is as follows, first import the model and prepare the code:
|
| 25 |
|
| 26 |
```python
|
|
|
|
| 75 |
|
| 76 |
|
| 77 |
|
| 78 |
+
### and more
|
| 79 |
+
|
| 80 |
However, in practical use, in order to achieve more diverse representations, it's recommended to do multiple inferences. Don't worry, it's really small so the inferences don't take much time, as follows:
|
| 81 |
|
| 82 |
```python
|
|
|
|
| 106 |
### 'Quick sort': 25,
|
| 107 |
### 'Recurrence': 2,
|
| 108 |
### 'Queue': 1
|
| 109 |
+
### }
|
| 110 |
```
|