Graph Positional and Structural Encoder · pyg-team/pytorch_geometric#8310

(21 comments) (0 reactions) (0 assignees)Python (3,514 forks)batch import

featuregood first issuehelp wantednn

Repository metrics

Stars: (19,985 stars)
PR merge metrics: (平均マージ 35d 1h) (30d で 14 merged PRs)

説明

🚀 The feature, motivation and pitch

It would be great to add the "GPSE" (Graph Positional and Structural Encoder) module to PyG, which acts as a trained encoder for Graph Transformers. Most importantly, we should support loading pre-trained encoders and provide them to the user, e.g., via

encoder = GPSE.from_pretrained("molpcba")

Code: https://github.com/G-Taxonomy-Workgroup/GPSE Paper: https://arxiv.org/abs/2307.07107 Twitter: https://twitter.com/RemyLau3/status/1717194030986190899

Any help on this is highly appreciated :)

Alternatives

No response

Additional context