许可证: mit

ReACC-py-retriever

这是为ReACC: 一个检索增强的代码补全框架开发的检索模型。

在论文中，该模型用于根据不完整的代码片段作为查询检索相似的代码。该模型也可用于不完整的代码到代码搜索、代码克隆检测。

py-retriever是一个类似BERT的编码器，由12个transformer层组成。它是在GraphCodeBERT的基础上，通过对比学习在Python编程语言中持续预训练的。更多细节可以在我们的论文中找到。

请注意，输入代码的格式与原始源代码不同。我们对源代码进行了规范化处理，以更好地从Python中的换行和缩进中捕获信息。一个输入的例子是：

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

要获取更多关于如何将源代码转换为此格式的信息，请参考ReACC GitHub仓库。