Reacc Py Retriever

🚀 ReACC-py-retriever

This is the retrieval model for ReACC: A Retrieval-Augmented Code Completion Framework, which can be used for code retrieval, incomplete code-to-code search, and code clone detection.

🚀 Quick Start

In the paper ReACC: A Retrieval-Augmented Code Completion Framework, this model is employed to retrieve similar codes when given an incomplete code snippet as a query. Besides, it can also be utilized for incomplete code-to-code search and code clone detection.

✨ Features

• Code Retrieval: Retrieve similar codes based on an incomplete code snippet.
• Incomplete Code-to-Code Search: Search for relevant complete codes from incomplete ones.
• Code Clone Detection: Detect code clones effectively.

📚 Documentation

Model Architecture

py-retriever is a BERT-like encoder composed of 12 transformer layers. It is continually pre-trained on GraphCodeBERT using contrastive learning in the Python programming language. For more detailed information, please refer to our paper.

Input Format

Note that the format of input codes differs from the original source code. We normalize the source codes to better capture information from line breaks and indentation in Python. Here is an example of the input:

sum = 0<endofline>for val in numbers:<endofline><INDENT>sum = sum+val

To learn more about how to convert source codes into this format, please refer to ReACC GitHub repo.

📄 License

This project is licensed under the MIT license.