公开(公告)号：CN119888350A

公开(公告)日：2025-04-25

申请号：CN202510042044.9

申请日：2025-01-10

Applicant: 湖南大学

Inventor： 李肯立 ,  刘诗语 ,  蔡宇辉 ,  周旭 ,  杨圣洪 ,  余思洋 ,  段明星 ,  吴帆 ,  秦云川

IPC: G06V10/764 ,  G06V10/774 ,  G06V10/82 ,  G06N3/042 ,  G06N3/047 ,  G06N3/0464 ,  G06N3/045 ,  G06N3/084

Abstract: 本申请实施例提供了一种多模型不平衡节点分类方法和装置。该方法包括：获取图数据集，所述图数据集包括训练集和测试集；基于所述训练集中的各节点信息，生成合成节点，以使得所述训练集中每个类别的节点数达到均衡分布；将生成的所有合成节点加入到所述训练集中形成新的训练集，并将所述新的训练集分成多个训练子集分别输入到图神经网络模型中的多个子模型中进行训练，直到所述图神经网络模型的性能收敛；将所述测试集输入至所述性能收敛的图神经网络模型中进行节点分类。本申请实施例的方法和装置，能够提高节点分类的性能。