pipeline_tag: 句子相似度 tags:

• sentence-transformers
• 特征提取
• 句子相似度
• transformers

mchochlov/codebert-base-cd-ft

这是一个基于sentence-transformers的模型：它将代码映射到768维的密集向量空间，并专门针对克隆检测任务进行了微调，采用了对比学习方法，训练数据来自BigCloneBench代码库的部分片段。

使用方式（Sentence-Transformers）

安装sentence-transformers后，可以轻松使用此模型：

pip install -U sentence-transformers

然后可以按如下方式使用模型：

from sentence_transformers import SentenceTransformer
code_fragments = [...]

model = SentenceTransformer('mchochlov/codebert-base-cd-ft')
embeddings = model.encode(code_fragments)
print(embeddings)