[发明专利]一种图神经网络数据采样方法、装置、设备及存储介质

摘要

本发明公开了一种图神经网络数据采样方法、装置、设备及存储介质；在本方案中，通过对原图数据集的各顶点进行聚类，并根据训练顶点的聚类类别进行排序的方式，使得采样进程在采样时，可在同一聚类内对一批训练顶点同时进行采样，从而提高采样的数据局部性；并且，由于同一个聚类内的训练顶点通常有更相似的属性，是紧密连接的部分，而不同聚类之间的连接很少，因此在同一个聚类内扩展的邻域顶点集中在相同的聚类内，相同聚类内的顶点在存储上是接近的，从而可以提高采样的数据局部性，并且限制邻域扩展的范围，提高子图采样的效率。