pyg-team/pytorch_geometric

Graph Positional and Structural Encoder

Offen

#8.310 geöffnet am 01.11.2023

 (21 Kommentare) (0 Reaktionen) (0 zugewiesene Personen)Python (3.514 Forks)batch import
featuregood first issuehelp wantednn

Repository-Metriken

Stars
 (19.985 Sterne)
PR-Merge-Metriken
 (Durchschn. Merge 35T 1h) (14 gemergte PRs in 30 T)

Beschreibung

🚀 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

No response

Contributor Guide