[发明专利]中文开放关系预测方法、装置、计算机设备和存储介质

权利要求书

1.一种中文开放关系预测方法，所述方法包括：

获取待预测句子中的命名实体，根据命名实体和所述待预测句子内容，构建实体对；所述实体对包括：头实体、头实体类型、尾实体、尾实体类型以及待预测句子内容；

当所述头实体类型和所述尾实体类型与预设的类型模式匹配，则将所述实体对中所述头实体、所述尾实体以及所述待预测句子内容输入关系预测模型，得到待预测句子中头实体与尾实体的关系类型；

所述关系预测模型为基于BERT的Att-Bi-GRU模型；

所述将所述实体对中所述头实体、所述尾实体以及所述待预测句子内容输入关系预测模型，得到待预测句子中头实体与尾实体的关系类型，包括：

将所述头实体、所述尾实体以及所述待预测句子内容输入所述Att-Bi-GRU模型，对所述头实体、所述尾实体以及所述待预测句子内容进行分词处理；

将分词处理得到的单词进行BERT嵌入，得到每个单词对应的词向量；

将所述词向量分别输入对应的Att-Bi-GRU网络，得到每个词向量的隐层状态；

对所述隐层状态进行组合，并通过输出层输出所述实体对对应的关系类型；

所述方法还包括：

确定单词t的特征向量为：

其中，h_T表示特征向量，h_i表示单词i对应的隐层状态，k表示总单词数；

α_i＝Aw_i+B

其中，set[head entity，tailentity]指的是位于头实体和位置之间的单词集合，A，B是网络的参数；

根据所述特征向量，确定单词t的权重。

2.根据权利要求1所述的方法，其特征在于，所述获取待预测句子中的命名实体，包括：

将待预测句子输入五级BIOSE标记系统，得到待预测句子的标记结果序列；

根据所述标记结果序列，确定待预测句子中的命名实体。

3.根据权利要求2所述的方法，其特征在于，根据命名实体和所述待预测句子内容，构建实体对，包括：

根据所述标记结果序列，确定所述标记结果序列中排序靠前的命名实体为头实体，排序靠后的命名实体为尾实体；

根据所述头实体的头实体类型和所述尾实体的尾实体类型，确定所述头实体和所述尾实体是否关联；

若是，则保存为实体对，若否，则从预先设置的实体对列表中删除。

4.根据权利要求1至3任一项所述的方法，其特征在于，在将所述实体对中所述头实体、所述尾实体以及所述待预测句子内容输入关系预测模型，得到待预测句子中头实体与尾实体的关系类型之后，所述方法还包括：

若所述关系预测模型无法预测所述实体对的关系类型，或者所述关系类型的概率对应的置信度小于预设值，则对待预测句子进行句子语义依存分析，得到实体对的关系类型。