基于双向编码表示转换的双模态软件分类模型

付晓峰,陈威岐,孙曜,潘宇泽

Bimodal software classification model based on bidirectional encoder representation from transformer

Xiaofeng FU,Weiqi CHEN,Yao SUN,Yuze PAN

表 1 双模态分类模型的参数设置

Tab.1 Parameter setting of bimodal classification model

参数 数值 迭代数 30 批数量 32 句子长度 代码512，文本256 学习率 10−5 隐藏层数 768 Dropout 0.5 CNN隐藏层数 损失函数 1 Cross-entropy loss 优化器 Adamax