Reacc Py Retriever

🚀 ReACC-py-retriever

このモデルは、ReACC: A Retrieval-Augmented Code Completion Framework の検索モデルです。

この論文では、このモデルは不完全なコードスニペットをクエリとして与えられた場合に、類似するコードを検索するために使用されます。また、不完全なコードからコードへの検索や、コードクローン検出にも使用できます。

py-retriever は、12層のトランスフォーマー層から構成されるBERTライクなエンコーダです。Pythonプログラミング言語で対照学習を用いて、GraphCodeBERT 上で継続的に事前学習されています。詳細については、当社の論文を参照してください。

入力コードの形式は、元のソースコードとは異なることに注意してください。Pythonの改行やインデントから情報をよりよく捉えるために、ソースコードを正規化しています。入力の例を以下に示します。

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

ソースコードをこの形式に変換する方法の詳細については、ReACC GitHubリポジトリ を参照してください。

📄 ライセンス

このプロジェクトはMITライセンスの下で公開されています。