[发明专利]对白角色预测方法、装置、电子设备及存储介质

权利要求书

1.一种对白角色预测方法，包括：

按照从头到尾的顺序遍历文本中的对白，将遍历到的每句对白分别作为待处理的对白；从所述文本中获取所述待处理的对白的上下文；

获取所述上下文中的上文中的各句文本内容的第一标签，所述第一标签为非对白或角色信息，所述角色为对白的说话人；

根据所述上下文及所述第一标签预测出所述对白的角色信息，包括：构建包含所述上下文、所述第一标签及所述待处理的对白的输入信息，将所述输入信息输入角色预测模型，得到预测出的所述待处理的对白的角色信息，所述角色预测模型为利用构建出的训练样本训练得到的，每条训练样本分别对应于文本中的一句对白，其中包括：所述对白对应的所述输入信息及第二标签，所述第二标签为所述对白的角色信息。

2.根据权利要求1所述的方法，还包括：

将被引号包围的文本内容作为对白；

和/或，针对任一句文本内容，利用预先训练得到的分类模型确定所述文本内容是否为对白。

3.一种对白角色预测装置，包括：

第一获取模块，用于按照从头到尾的顺序遍历文本中的对白，将遍历到的每句对白分别作为待处理的对白；从所述文本中获取所述待处理的对白的上下文；

第二获取模块，用于获取所述上下文中的上文中的各句文本内容的第一标签，所述第一标签为非对白或角色信息，所述角色为对白的说话人；

预测模块，用于根据所述上下文及所述第一标签预测出所述对白的角色信息，包括：构建包含所述上下文、所述第一标签及所述待处理的对白的输入信息，将所述输入信息输入角色预测模型，得到预测出的所述待处理的对白的角色信息，所述角色预测模型为利用构建出的训练样本训练得到的，每条训练样本分别对应于文本中的一句对白，其中包括：所述对白对应的所述输入信息及第二标签，所述第二标签为所述对白的角色信息。

4.根据权利要求3所述的装置，其中，

所述第一获取模块进一步用于，将被引号包围的文本内容作为对白，和/或，针对任一句文本内容，利用预先训练得到的分类模型确定所述文本内容是否为对白。

5.一种电子设备，包括：

至少一个处理器；以及

与所述至少一个处理器通信连接的存储器；其中，

所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行权利要求1-2中任一项所述的方法。

6.一种存储有计算机指令的非瞬时计算机可读存储介质，其中，所述计算机指令用于使所述计算机执行权利要求1-2中任一项所述的方法。