RepoFusion
Data
\nStack-Repo can be accessed via the Datasets section of this space. Please see the README for complete details.
\nTrained Checkpoints
\nThe trained checkpoints can be downloaded from the Models. Please see the README for complete details.
\nCode
\nThe code for training and evaluating RepoFusion, finetuning CodeT5, and details of how to run the scripts can be found here
\nCitation
\n@article{shrivastava2023repofusion,\n title={RepoFusion: Training Code Models to Understand Your Repository},\n author={Shrivastava, Disha and Kocetkov, Denis and de Vries, Harm and Bahdanau, Dzmitry and Scholak, Torsten},\n journal={arXiv preprint arXiv:2306.10998},\n year={2023}\n}\n\n","classNames":"hf-sanitized hf-sanitized-eXu2ItvB9sfHHfFNnzpAK"},"users":[{"_id":"62dea51feee79ca5ef8f834a","avatarUrl":"/avatars/17e9356625785622762874fe96900599.svg","isPro":false,"fullname":"Denis Kocetkov","user":"denisko","type":"user"},{"_id":"6438284b819f3ab20d185a3b","avatarUrl":"https://cdn-avatars.huggingface.co/v1/production/uploads/6438284b819f3ab20d185a3b/b29SoH7_Bji6GfCszjqOE.jpeg","isPro":false,"fullname":"Disha Shrivastava","user":"shrivasd","type":"user"},{"_id":"62ce91e8b5811ba53c0ce7b9","avatarUrl":"https://cdn-avatars.huggingface.co/v1/production/uploads/62ce91e8b5811ba53c0ce7b9/ZNgd-p0h-q1j374QqEGrb.jpeg","isPro":false,"fullname":"Harm de Vries","user":"harmdevries","type":"user"},{"_id":"61bcebe9351aea330acb5b1f","avatarUrl":"/avatars/d054e80e3b1635c0f5664c75e52d5490.svg","isPro":false,"fullname":"Bahdanau","user":"Dzmitry","type":"user"},{"_id":"60ecaa5efee13fee7ada7af4","avatarUrl":"https://cdn-avatars.huggingface.co/v1/production/uploads/1637694419284-60ecaa5efee13fee7ada7af4.jpeg","isPro":false,"fullname":"Torsten Scholak","user":"tscholak","type":"user"}],"userCount":5,"collections":[],"datasets":[{"author":"RepoFusion","downloads":747,"gated":false,"id":"RepoFusion/Stack-Repo","lastModified":"2023-07-10T19:43:46.000Z","private":false,"repoType":"dataset","likes":9,"isLikedByUser":false,"isBenchmark":false}],"models":[{"author":"RepoFusion","authorData":{"_id":"6489e6aab89b3a67ba8a5ee8","avatarUrl":"https://www.gravatar.com/avatar/f543fc72b9842b8b765e567a6858b564?d=retro&size=100","fullname":"RepoFusion","name":"RepoFusion","type":"org","isHf":false,"isHfAdmin":false,"isMod":false,"followerCount":25,"isUserFollowing":false},"downloads":0,"gated":false,"id":"RepoFusion/trained_checkpoints","availableInferenceProviders":[],"lastModified":"2023-06-26T14:57:56.000Z","likes":9,"private":false,"repoType":"model","isLikedByUser":false}],"paperPreviews":[],"spaces":[],"buckets":[],"numBuckets":0,"numDatasets":1,"numModels":1,"numSpaces":1,"lastOrgActivities":[],"acceptLanguages":["*"],"canReadRepos":false,"canReadSpaces":false,"blogPosts":[],"currentRepoPage":0,"filters":{},"paperView":false}">
AI & ML interests
None defined yet.
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
The code for training and evaluating RepoFusion, finetuning CodeT5, and details of how to run the scripts can be found here
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}
}