-
公开(公告)号:CN120010821A
公开(公告)日:2025-05-16
申请号:CN202510081504.9
申请日:2025-01-17
Applicant: 支付宝(杭州)信息技术有限公司
Abstract: 本说明书实施例提供一种代码嵌入模型的训练方法,包括:利用编码模块处理目标输入,得到其中各字符的输出向量。利用文本域代码共享的聚合模块,基于第一锚点句向量,对各字符的输出向量进行基于注意力机制的第一聚合处理,得到目标输入的第一特征向量。利用目标输入的类型专用的目标聚合模块,分别基于该目标聚合模块对应的目标数目的第二锚点句向量,对各字符的输出向量进行基于注意力机制的第二聚合处理,得到目标输入的第二特征向量。基于样本集中各文本和各代码片段各自的第一和第二特征向量,确定预测损失,并基于其更新第一和第二聚合模块以及第一和第二锚点句向量。