图神经网络训练方法、标签生成方法、装置、设备及介质
申请号:CN202410921639
申请日期:2024-07-10
公开号:CN118780318A
公开日期:2024-10-15
类型:发明专利
摘要
本申请提供了一种图神经网络训练方法、标签生成方法、装置、设备及介质,该方法应用于图数据技术领域,该方法包括:获取训练图数据中第i个节点数据的目标邻域结构与所述第i个节点数据的样本标签;基于所述目标邻域结构中的邻居节点数据的原始节点特征,对所述第i个节点数据的原始节点特征进行特征聚合,得到所述第i个节点数据的平滑特征;将所述第i个节点数据的平滑特征输入至待训练的图神经网络,输出所述第i个节点数据的目标预测标签;基于所述第i个节点数据的目标预测标签和所述样本标签之间的差异,对所述待训练的图神经网络进行迭代训练,得到训练后的图神经网络。该方法能够适用对不同图结构中节点的标签进行预测。
技术关键词
邻域
节点特征
数据
神经网络训练方法
标签生成方法
邻居
可执行程序代码
神经网络训练装置
标签生成装置
样本
电子设备
超参数
可读存储介质
模块
存储器
计算机
矩阵