一种基于预训练的多类型代码自动生成方法、装置及介质

基本信息

申请号 CN202110102379.7 申请日 -
公开(公告)号 CN112799655A 公开(公告)日 2021-05-14
申请公布号 CN112799655A 申请公布日 2021-05-14
分类号 G06F8/34;G06F8/38;G06F8/41 分类 计算;推算;计数;
发明人 孙晓飞;孟昱先 申请(专利权)人 浙江香侬慧语科技有限责任公司
代理机构 北京国科程知识产权代理事务所(普通合伙) 代理人 曹晓斐
地址 311121 浙江省杭州市余杭区余杭街道文一西路1818-2号9幢903室
法律状态 -

摘要

摘要 本发明提供一种基于预训练的多类型代码自动生成方法、装置及介质,属于文本生成领域。本发明主要包括利用互联网上的代码数据对文本生成模型进行预训练得到自动代码生成模型的步骤;向自动代码生成模型中输入目标代码功能描述和目标代码语言类型,利用自动代码生成模型的代码编码器得到代码功能语义特征表示的步骤;以及利用代码功能语义表示通过自动代码生成模型的代码解码器自动生成目标代码的步骤。本发明的有益效果是,利用了互联网上现成的海量代码数据资料,通过预训练技术提高文本生成模型生成代码的精度,自动生成多种语言类型的代码,可以极大的提高写代码的效率以及实现代码自动纠错。