spamosaic.architectures.wlgcn.WLGCN_vanilla

class spamosaic.architectures.wlgcn.WLGCN_vanilla(*args: Any, **kwargs: Any)

Parameter-free LightGCN-style propagation.

Accepts a precomputed normalized sparse adjacency \(\hat{A} \approx D^{-1/2}(A + I\cdot\text{fill})D^{-1/2}\) as a torch_sparse.SparseTensor.

Parameters:

• K (int, default=1) – Number of propagation steps (returns K+1 representations including the 0-th order input).

• agg ({'cat', 'sum', 'mean'}, default='cat') – How to aggregate representations across steps.