摘要
一种威胁情报抽取模型训练、威胁情报分类的方法及装置,该威胁情报抽取模型训练的方法包括:获取基础模型;使用AdamW优化器和学习率调整策略,将数据加载器处理后的数据作为样本,按批次传递给基础模型进行训练,得到最终模型,其中,所述样本以威胁情报作为输入,以漏洞类型的描述信息作为输出。通过本发明实施例提供的方法及装置,提升了模型的训练效率和训练效果,并提高了最终的分类效果。
技术关键词
前馈神经网络
漏洞
序列
加载器
注意力机制
解码器
基础
样本
数据
训练集
标签
编码器
搜索算法
策略
超参数
处理器
可读存储介质
指令