mbahrami commited on
Commit
6aa1d2b
1 Parent(s): e584b28

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -2
README.md CHANGED
@@ -12,6 +12,15 @@ tokenizer = AutoTokenizer.from_pretrained("Fujitsu/AugCode")
12
  model = AutoModelForSequenceClassification.from_pretrained("Fujitsu/AugCode")
13
 
14
  ```
15
- Then you may use it to infer the similarity between a given docstring and code.
16
-
17
 
 
 
 
 
 
 
 
 
 
 
12
  model = AutoModelForSequenceClassification.from_pretrained("Fujitsu/AugCode")
13
 
14
  ```
15
+ Then you may use `model` to infer the similarity between a given docstring and code.
 
16
 
17
+ ### Citation
18
+ ```bibtex@misc{bahrami2021augcode,
19
+ title={AugmentedCode: Examining the Effects of Natural Language Resources in Code Retrieval Models},
20
+ author={Mehdi Bahramix, N. C. Shrikanthy, Yuji Mizobuchiz, Lei Liux, Masahiro Fukuyoriz, Wei-Peng Chenx, Kazuki Munakataz},
21
+ year={2021},
22
+ eprint={TBA},
23
+ archivePrefix={TBA},
24
+ primaryClass={cs.CL}
25
+ }
26
+ ```