torch_geometric.transforms.HalfHop
- class HalfHop(alpha: float = 0.5, p: float = 1.0)[source]
Bases:
BaseTransform
The graph upsampling augmentation from the “Half-Hop: A Graph Upsampling Approach for Slowing Down Message Passing” paper. The graph is augmented by adding artificial slow nodes between neighbors to slow down message propagation. (functional name:
half_hop
).Note
HalfHop
augmentation is not supported ifdata
hasedge_weight
oredge_attr
.- Parameters:
import torch_geometric.transforms as T transform = T.HalfHop(alpha=0.5) data = transform(data) # Apply transformation. out = model(data.x, data.edge_index) # Feed-forward. out = out[~data.slow_node_mask] # Get rid of slow nodes.