Maninder Singh
commited on
Commit
·
bc528bd
1
Parent(s):
8953341
Update README.md
Browse files
README.md
CHANGED
|
@@ -14,12 +14,12 @@ tags:
|
|
| 14 |
- code-classification
|
| 15 |
base_model: huggingface/CodeBERTa-small-v1
|
| 16 |
---
|
| 17 |
-
This Model is a fine-tuned version of *huggingface/CodeBERTa-small-v1* on *cakiki/rosetta-code* Dataset for
|
| 18 |
## Training Details:
|
| 19 |
Model is trained for 25 epochs on Azure for nearly 26000 Datapoints for above Mentioned 26 Programming Languages<br> extracted from Dataset having 1006 of total Programming Language.
|
| 20 |
### Programming Languages this model is able to detect vs Examples used for training
|
| 21 |
<ol>
|
| 22 |
-
<li>'ARM Assembly'
|
| 23 |
<li>'AppleScript'</li>
|
| 24 |
<li>'C'</li>
|
| 25 |
<li>'C#'</li>
|
|
@@ -71,7 +71,7 @@ Model is trained for 25 epochs on Azure for nearly 26000 Datapoints for above Me
|
|
| 71 |
|
| 72 |
## Inference Code
|
| 73 |
|
| 74 |
-
```
|
| 75 |
import torch
|
| 76 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline
|
| 77 |
model_name = 'Maninder120996/programming-language-identification'
|
|
|
|
| 14 |
- code-classification
|
| 15 |
base_model: huggingface/CodeBERTa-small-v1
|
| 16 |
---
|
| 17 |
+
This Model is a fine-tuned version of *huggingface/CodeBERTa-small-v1* on *cakiki/rosetta-code* Dataset for 26 Programming Languages as mentioned below.
|
| 18 |
## Training Details:
|
| 19 |
Model is trained for 25 epochs on Azure for nearly 26000 Datapoints for above Mentioned 26 Programming Languages<br> extracted from Dataset having 1006 of total Programming Language.
|
| 20 |
### Programming Languages this model is able to detect vs Examples used for training
|
| 21 |
<ol>
|
| 22 |
+
<li>'ARM Assembly':</li>
|
| 23 |
<li>'AppleScript'</li>
|
| 24 |
<li>'C'</li>
|
| 25 |
<li>'C#'</li>
|
|
|
|
| 71 |
|
| 72 |
## Inference Code
|
| 73 |
|
| 74 |
+
```python
|
| 75 |
import torch
|
| 76 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline
|
| 77 |
model_name = 'Maninder120996/programming-language-identification'
|