| license: mit | |
| This is an example model trained on C declarations. It is a foundational model that can essentially only complete C declarations. | |
| It has been trained on a variety of OSS software declarations, not entire functions. | |
| Usage: | |
| ```python | |
| # make a prediction | |
| from transformers import pipeline | |
| from pprint import pprint | |
| fill_mask = pipeline( | |
| "fill-mask", | |
| model="mstaron/declBERTa", | |
| tokenizer="mstaron/declBERTa" | |
| ) | |
| strPredicted = fill_mask("int i = <mask>;", top_k=10) | |
| pprint(strPredicted) | |
| ``` |