|
--- |
|
language: |
|
- java |
|
- code |
|
license: apache-2.0 |
|
widget: |
|
- text: 'public [MASK] isOdd(Integer num){if (num % 2 == 0) {return "even";} else {return "odd";}}' |
|
--- |
|
## JavaBERT |
|
A BERT-like model pretrained on Java software code. |
|
|
|
### Training Data |
|
The model was trained on 2,998,345 Java files retrieved from open source projects on GitHub. A ```bert-base-uncased``` tokenizer is used by this model. |
|
|
|
### Training Objective |
|
A MLM (Masked Language Model) objective was used to train this model. |
|
|
|
### Usage |
|
```python |
|
from transformers import pipeline |
|
pipe = pipeline('fill-mask', model='CAUKiel/JavaBERT') |
|
output = pipe(CODE) # Replace with Java code; Use '[MASK]' to mask tokens/words in the code. |
|
``` |
