torch_geometric.transforms.RootedEgoNets

class RootedEgoNets(num_hops: int)[source]

Bases: RootedSubgraph

收集图中每个节点的根\(k\)-跳EgoNets,如“从星到子图:通过局部结构感知提升任何GNN”论文中所述。

Parameters:

num_hops (int) – 跳数 \(k\)