带角色的文本分类模型以及对话文本分类方法及装置

摘要

提供带角色的文本分类模型以及对话文本分类方法及装置。该模型包括：语句编码层、语句匹配层、语句分类层和对话分类层；对话文本经过语句编码层，其中的多个语句被编码为多个语句向量；多个语句向量经过语句匹配层，被分别归类为至少两个角色，构建每个语句向量的上下文语义表示，并基于每个语句向量及其对应的上下文语义表示构建上下文增强的语句向量；上下文增强的语句向量经过语句分类层，与相应的角色向量进行拼接，得到带角色信息的语句向量，并得到对应的风险预测结果；所有语句向量的风险预测结果经过对话分类层，被聚合得到对话文本的风险预测结果。该模型将对话文本中的角色和语句关联建模，从而得到与角色相关的风险预测结果。