一种基于图神经网络节点特征传播优化的文本分类方法
Abstract:
本发明提出一种基于图神经网络节点特征传播优化的文本分类方法,解决了现有文本分类方法不能兼顾分类准确度和分类效率的问题,包括:获取文本数据集,构成以文本节点、单词节点为基础的二元异构图,利用二元异构图形成对应的邻接矩阵A;将邻接矩阵A拆解,构建图神经网络节点特征传播优化模型;对图神经网络节点特征传播优化模型进行训练;利用训练好的图神经网络节点特征传播优化模型对文本数据集进行分类。本发明将文本分类问题转换为节点分类问题,在保持节点分类的准确度下,提高节点特征传播的效率,从而提高文本分类的速度,同时减少文本标注数量,从而减少标注成本。
Patent Agency Ranking
0/0