| | --- |
| | library_name: transformers.js |
| | base_model: |
| | - neulab/codebert-javascript |
| | pipeline_tag: fill-mask |
| | --- |
| | |
| |
|
| |
|
| | # codebert-javascript (ONNX) |
| |
|
| |
|
| | This is an ONNX version of [neulab/codebert-javascript](https://huggingface.co/neulab/codebert-javascript). It was automatically converted and uploaded using [this Hugging Face Space](https://huggingface.co/spaces/onnx-community/convert-to-onnx). |
| |
|
| |
|
| | ## Usage with Transformers.js |
| |
|
| |
|
| | See the pipeline documentation for `fill-mask`: https://huggingface.co/docs/transformers.js/api/pipelines#module_pipelines.FillMaskPipeline |
| | |
| | |
| | --- |
| | |
| | |
| | This is a `microsoft/codebert-base-mlm` model, trained for 1,000,000 steps (with `batch_size=32`) on **JavaScript** code from the `codeparrot/github-code-clean` dataset, on the masked-language-modeling task. |
| |
|
| | It is intended to be used in CodeBERTScore: [https://github.com/neulab/code-bert-score](https://github.com/neulab/code-bert-score), but can be used for any other model or task. |
| |
|
| | For more information, see: [https://github.com/neulab/code-bert-score](https://github.com/neulab/code-bert-score) |
| |
|
| | ## Citation |
| |
|
| | If you use this model for research, please cite: |
| | ``` |
| | @article{zhou2023codebertscore, |
| | url = {https://arxiv.org/abs/2302.05527}, |
| | author = {Zhou, Shuyan and Alon, Uri and Agarwal, Sumit and Neubig, Graham}, |
| | title = {CodeBERTScore: Evaluating Code Generation with Pretrained Models of Code}, |
| | publisher = {arXiv}, |
| | year = {2023}, |
| | } |
| | ``` |
| |
|