摘要
本申请公开了一种图结构学习模型训练方法及装置,将获取的原始图结构的初始节点特征矩阵和初始邻接矩阵输入到多层图卷积网络中,得到第一节点特征矩阵,并基于此构建第二邻接矩阵;基于第二邻接矩阵和最后一层图卷积网络对应的第一节点特征矩阵,得到第二节点特征矩阵;基于第二节点特征矩阵计算得到第一权重矩阵和相似度矩阵,并基于第二邻接矩阵、权重矩阵和相似度矩阵,计算得到第三邻接矩阵;对第三邻接矩阵进行加权求和,得到目标邻接矩阵;基于目标邻接矩阵和最后一层图卷积网络对应的第一节点特征矩阵对多层图卷积网络进行训练。通过优化图结构提高图结构数据分类的准确性。
技术关键词
学习模型训练方法
节点特征
网络
矩阵
模型训练装置
可读存储介质
近邻算法
预测类别
处理器
数据分类
输入模块
存储器
计算机
电子设备