该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

Rewriting the Code: A Simple Framework for Large Language Model Augmented Semantic Code Search

This repo contains code for Rewriting the Code: A Simple Framework for Large Language Model Augmented Semantic Code Search, accepted to ACL 2024. In this codebase we provide instructions for reproducing our results from the paper. We hope that this work can be useful for future research on Generation-Augmented Retrieval framework for code search.

Environment

conda create -n ReCo python=3.8 -y
conda activate ReCo
conda install pytorch-gpu=1.7.1 -y
pip install transformers datasets tqdm tree-sitter openai fairscale
fire sentencepiece backoff edit_distance pyserini

Data

For the detailed information of data we used in our experiments, please refer to README.md in ./data.

ReCo

For the detailed information of ReCo and GAR in our paper, please refer to README.md in ./ReCo.

Metrics

For the detailed information of Code Style Distance in our paper, please refer to README.md in ./metrics.

Citation

If you found this repository useful, please consider citing:

@article{li2024rewriting,
  title={Rewriting the Code: A Simple Method for Large Language Model Augmented Code Search},
  author={Li, Haochen and Zhou, Xin and Shen, Zhiqi},
  journal={arXiv preprint arXiv:2401.04514},
  year={2024}
}

空文件

简介

暂无描述 展开 收起
Python 等 2 种语言
取消

发行版

暂无发行版

贡献者

全部

近期动态

不能加载更多了
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化