torch_geometric.transforms.RootedRWSubgraph

class RootedRWSubgraph(walk_length: int, repeat: int = 1)[source]

基础类:RootedSubgraph

为图中的每个节点收集基于根随机游走的子图,如“从星到子图:通过局部结构感知提升任何GNN”论文中所述。

Parameters:
  • walk_length (int) – 随机游走的长度。

  • repeat (int, optional) – 重复随机游走的次数以减少随机性。(默认值:1