一种融合实体类型的关系三元组抽取方法
摘要:
本发明公开了一种融合实体类型的关系三元组抽取方法,包括以下步骤:收集文本数据作为训练样本;对收集的训练样本数据进行清洗获得数据集;切分数据集并将其按照一定比例划分为训练集、验证集与测试集;搭建基于BERT预训练模型的深度学习网络并加载预训练参数获得训练样本深度表达;在模型后搭建Fast Gradient Method对抗网络提升模型的鲁棒和泛化性能;搭建多头注意力机制与深度神经网络的关系三元组提取模型;进行模型的训练与测试,对验证集采用K折交叉验证的方法保存K折模型,综合K折模型对测试集进行测试,将平均概率作为模型的测试结果;输出模型AttnFGM‑MARE。
公开/授权文献
0/0