shrivasd commited on
Commit
7c450cb
1 Parent(s): 05461e9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -1
README.md CHANGED
@@ -1,5 +1,13 @@
1
  ---
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
4
 
5
- These versions of the model weights are strictly permitted for use exclusively in conjunction with the review process for the paper with Submission Number 13449. Upon completion of the review process, a de-anonymized version of the model weights will be released under appropriate license.
 
1
  ---
2
  license: mit
3
  ---
4
+ This repository contains the trained checkpoints corresponding to our paper [RepoFusion: Training Code Models to Understand Your Repository](). The released checkpoints are:
5
+ - `RepoFusion_PPC`: RepoFusion model trained with prompt proposal repo contexts. This is our best-performing model.
6
+ - `RepoFusion_BM25`: RepoFusion model trained with BM25 repo contexts.
7
+ - `RepoFusion_RandomNN`: RepoFusion model trained with RandomNN repo contexts.
8
+ - `finetuned_codet5base_512`: Our finetuned CodeT5-base model. This was used as initialization for our RepoFusion models.
9
+ - `finetuned_codet5large_512`: Our finetuned CodeT5-large model. This was used as a baseline.
10
+
11
+ For details of how these models were trained and evaluated, please check our paper [RepoFusion: Training Code Models to Understand Your Repository]().
12
 
13
+ ## Citation