metadata

title: README
emoji: 📉
colorFrom: yellow
colorTo: indigo
sdk: static
pinned: false

RepoFusion: Training Code Models to Understand Your Repository

Disha Shrivastava, Denis Kocetkov, Harm de Vries, Dzmitry Bahdanau, Torsten Scholak

This space contains the released resources for our paper RepoFusion: Training Code Models to Understand Your Repository. A block diagram of our approach can be found below. For more details, refer to the paper.

Data

Stack-Repo can be accessed via the Datasets section of this space. Please see the README for complete details.

Trained Checkpoints

The trained checkpoints can be downloaded from the Models. Please see the README for complete details.

Code

We will be releasing the code for training and evaluating RepoFusion, finetuning CodeT5, and details of running the scripts shortly. Watch out this space for updates.

Citation

@article{shrivastava2023repofusion,
  title={RepoFusion: Training Code Models to Understand Your Repository},
  author={Shrivastava, Disha and Kocetkov, Denis and de Vries, Harm and Bahdanau, Dzmitry and Scholak, Torsten},
  journal={arXiv preprint arXiv:2306.10998},
  year={2023}
}