search

pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch
https://pyg.org
MIT License
20.52k stars 3.57k forks source link

Add pure SSG with no lin #9402

Closed Kh4L closed 2 weeks ago

Kh4L commented 3 weeks ago

Recent work in link prediction tasks (e.g., https://arxiv.org/abs/2007.00970) has shown the need for a pure GCN convolution, without applying any linearity to the output. This PR modifies the existing SGConv to support GCN convolution.

Kh4L commented 2 weeks ago

Closing after offline convo w @rusty1s : https://pytorch-geometric.readthedocs.io/en/latest/generated/torch_geometric.nn.conv.LGConv.html exists