一种用于机械故障诊断知识图谱构造的实体抽取方法

本发明公开了一种用于机械故障诊断知识图谱构造的实体抽取方法，包括步骤S1：构建机械故障诊断语料数据库，并将其划分为训练集和测试集；步骤S2：构建融合BERT‑BiLSTM‑CRF、数据增强和对抗训练的机械故障诊断知识图谱命名实体识别模型EBBC‑DAM‑PATM，并对其进行训练测试，得到训练后的EBBC‑DAM‑PATM模型；步骤S3：利用训练好的EBBC‑DAM‑PATM模型进行命名实体识别；步骤S4：利用neo4j图数据库将提取得到的实体字构建成知识图谱。本发明所述的数据增强模块可以依据数据量自适应完成，减轻样本不平衡的问题并缓解模型的过拟合。