torch_geometric.transforms.RootedEgoNets

class RootedEgoNets(num_hops: int)[source]

Bases: RootedSubgraph

Collects rooted \(k\)-hop EgoNets for each node in the graph, as described in the “From Stars to Subgraphs: Uplifting Any GNN with Local Structure Awareness” paper.

Parameters:

num_hops (int) – the number of hops \(k\).