torch_geometric.transforms.RootedRWSubgraph

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

Bases: RootedSubgraph

Collects rooted random-walk based subgraphs for each node in the graph, as described in the “From Stars to Subgraphs: Uplifting Any GNN with Local Structure Awareness” paper.

Parameters:

walk_length (int) – the length of the random walk.
repeat (int, optional) – The number of times of repeating the random walk to reduce randomness. (default: 1)

Read the Docs v: stable

Versions: latest; stable; 2.5.3; 2.5.2; 2.5.1; 2.5.0; 2.4.0; 2.3.1; 2.3.0; 2.2.0; 2.1.0; 2.0.4; 2.0.3; 2.0.2; 2.0.1; 2.0.0; 1.7.2; 1.7.1; 1.7.0; 1.6.3; 1.6.2; 1.6.1; 1.6.0; 1.5.0; 1.4.3; 1.4.2; 1.4.1; 1.3.2; 1.3.1; 1.3.0

Downloads

On Read the Docs: Project Home; Builds