File size: 844 Bytes
4d7dfec 05461e9 4d7dfec 7c450cb b90a508 7c450cb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
---
license: mit
---
This repository contains the trained checkpoints corresponding to our paper [RepoFusion: Training Code Models to Understand Your Repository](). The released checkpoints are:
- `RepoFusion_PPC`: RepoFusion model trained with prompt proposal repo contexts. This is our best-performing model.
- `RepoFusion_BM25`: RepoFusion model trained with BM25 repo contexts.
- `RepoFusion_RandomNN`: RepoFusion model trained with RandomNN repo contexts.
- `finetuned_codet5base_512`: Our finetuned CodeT5-base model. This was used as initialization for our RepoFusion models.
- `finetuned_codet5large_512`: Our finetuned CodeT5-large model. This was used as a baseline.
For details of how these models were trained and evaluated, please check our paper [RepoFusion: Training Code Models to Understand Your Repository]().
## Citation
|