| --- |
| 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}, |
| } |
| ``` |
|
|