一种地球科学地质实体关系抽取方法、介质、设备

本发明提出一种地球科学地质实体关系抽取方法、介质、设备，涉及实体关系抽取技术领域，方法包括：将地球科学地质数据、实体以及标签划分为训练集和测试集；构建地质实体关系抽取的神经网络模型，将训练集输入模型进行一阶段训练；使用训练集对一阶段训练后的模型进行二阶段训练，以二阶段训练模型的输入作为状态，输出为动作概率分布，并设置奖励函数和优势函数，构建PPO模型对二阶段训练模型进行优化；根据一阶段训练的损失函数和二阶段训练时PPO模型的损失函数，构建联合损失函数；使用测试集对二阶段训练后的模型进行测试，使用测试后的模型进行地球科学地质实体关系抽取。本发明有效解决了训练中的不稳定性与局部最优问题。