🔥 AugCode 🔥

This is Augmented Code Model which is a fined-tune model of CodeBERT 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.

from transformers import AutoTokenizer, AutoModel

tokenizer = AutoTokenizer.from_pretrained("microsoft/codebert-base")

model = AutoModel.from_pretrained("microsoft/codebert-base")

Then you may use it to infer the similarity between a given docstring and code.