[发明专利]一种确定事件主体的方法和装置

说明书

本发明公开了一种确定事件主体的方法和装置，涉及计算机技术领域。该方法的一具体实施方式包括：构建训练样本集；其中，所述训练样本集中包括：多个训练样本；所述训练样本中包括：训练文本、所述训练文本的事件类型和所述训练文本的事件主体；根据所述训练样本集训练抽取模型；其中，所述抽取模型包括：输入层、注意力层、序列层和输出层；确定预测文本的事件类型；根据所述预测文本的事件类型和训练后的抽取模型，确定所述预测文本的事件主体。通过本发明实施例能够得到具有更高准确性和可靠性的事件主体。

技术领域

本发明涉及计算机技术领域，尤其涉及一种确定事件主体的方法和装置。

背景技术

事件主体可以用于识别风险、辅助决策以及确定事件的发展趋势等。在实际应用场景中，可以根据文本确定事件主体。

现有技术中，采用word2vec对文本进行处理，以确定文本的事件主体。

但是，在实现本发明过程中，发明人发现该方法捕捉文本的语义信息的能力较差，使得到的事件主体的准确性和可靠性较低。

发明内容

有鉴于此，本发明实施例提供一种确定事件主体的方法和装置，能够得到具有更高准确性和可靠性的事件主体。

第一方面，本发明实施例提供了一种确定事件主体的方法，包括：

构建训练样本集；其中，所述训练样本集中包括：多个训练样本；所述训练样本中包括：训练文本、所述训练文本的事件类型和所述训练文本的事件主体；

根据所述训练样本集训练抽取模型；其中，所述抽取模型包括：输入层、注意力层、序列层和输出层；

确定预测文本的事件类型；

根据所述预测文本的事件类型和训练后的抽取模型，确定所述预测文本的事件主体。

可选地，

所述构建训练样本集，包括：

获取所述训练文本、所述训练文本的事件类型和所述训练文本的事件主体；

根据预设的字符规则对所述训练文本、所述训练文本的事件类型和事件主体进行处理，处理后的所述训练文本、所述训练文本的事件类型和事件主体构成所述训练样本。

可选地，

所述训练样本包括：训练文本、所述训练文本的事件类型和事件主体；

所述根据所述训练样本集训练抽取模型，包括：

将所述训练文本、所述训练文本的事件类型和事件主体输入所述抽取模型，得到所述训练文本中的字符作为事件主体的概率和所述训练文本的事件类型中的字符作为事件主体的概率；

根据所述训练文本中的字符作为事件主体的概率和所述训练文本的事件类型中的字符作为事件主体的概率，调整所述抽取模型的参数。

可选地，

所述将所述训练文本、所述训练文本的事件类型和事件主体输入所述抽取模型，得到所述训练文本中的字符作为事件主体的概率和所述训练文本的事件类型中的字符作为事件主体的概率，包括：

将所述训练文本、所述训练文本的事件类型和事件主体输入所述输入层，得到训练文本编码、所述训练文本的事件类型编码和事件主体编码；

将携带所述训练文本的事件主体编码的所述训练文本编码和所述训练文本的事件类型编码输入所述注意力层，得到若干语义向量；其中，所述若干语义向量包括：与所述训练文本中的字符对应的语义向量和与所述训练文本的事件类型中的字符对应的语义向量；

将所述若干语义向量输入所述序列层，得到若干序列向量；