pyg-team/pytorch_geometric

Graph Positional and Structural Encoder

開放

#8,310 建立於 2023年11月1日

 (21 則留言) (0 個反應) (0 位負責人)Python (3,514 個分叉)batch import
featuregood first issuehelp wantednn

倉庫指標

星標
 (19,985 顆星)
PR 合併指標
 (平均合併 35天 1小時) (30 天內合併 14 個 PR)

描述

🚀 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

貢獻者指南