subgraphs_learning 源码

subgraphs_learning 给定一个非常大的图,其中包含少量标记的顶点,我们希望能够仅在大小受限的子图上进行预训练,从而标记给定的一个或多个节点。 要运行代码,请运行文件subgraphs_GCN_reg.py。 该文件可以运行以下命令: 小型数据集(Cora,CiteSeer,Pubmed,CS)。 来自GraphSaint git的大型数据集。 为了运行大型数据集,请在主Graph_Saint选项中将其更改为True。 否则,您将运行小型数据集,您可以在其中选择以下输入之一: BOW(更改BOW =主输入为True) 2k向量输入(Bow = False)。 此功能表示训练集中属于每个类别的第一和第二邻居的数量。 例如,假设具有3个可能标签的分类任务,以及一个具有5个邻居和30个第二邻居的节点。 进一步假设1个第一邻居属于训练集并具有标签A,3个第二邻居属于训练