pyg-team/pytorch_geometric

Graph Positional and Structural Encoder

Open

#8310 aperta il 1 nov 2023

Vedi su GitHub
 (21 commenti) (0 reazioni) (0 assegnatari)Python (3514 fork)batch import
featuregood first issuehelp wantednn

Metriche repository

Star
 (19.985 star)
Metriche merge PR
 (Merge medio 16g 3h) (13 PR mergiate in 30 g)

Descrizione

🚀 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

Guida contributor

Direzione di ricerca
Studia il paper GPSE e l'implementazione di riferimento, quindi integra il meccanismo di caricamento dell'encoder pre addestrato nello zoo dei modelli di PyG.
Tech stack
pythonpytorch
Dominio
machine learningapi
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
Mezza giornata
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
PythonPyTorchGit
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
50