File size: 686 Bytes
f256354 e584b28 f256354 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# 🔥 AugCode 🔥
This is Augmented Code Model which is a fined-tune model of [CodeBERT](https://huggingface.co/microsoft/codebert-base) for processing of similarity between given docstring and code. This model is fined-model based on Augmented Code Corpus with ACS=4.
## How to use the model ?
Similar to other huggingface model, you may load the model as follows.
```python
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Fujitsu/AugCode")
model = AutoModelForSequenceClassification.from_pretrained("Fujitsu/AugCode")
```
Then you may use it to infer the similarity between a given docstring and code.
|