pyg-team/pytorch_geometric

Add `KerGNN` to `conv.nn` as layer type

Open

#4937 aperta il 7 lug 2022

Vedi su GitHub
 (4 commenti) (0 reazioni) (0 assegnatari)Python (3514 fork)batch import
0 - Priority P0featurehelp wantednn

Metriche repository

Star
 (19.985 star)
Metriche merge PR
 (Merge medio 35g 1h) (14 PR mergiate in 30 g)

Descrizione

🚀 The feature, motivation and pitch

Dear PyTorch-Geometric team, dear Matthias,

stumbled upon this novel GNN layer type idea named KerGNN, and it worked out quite well in their and my own experiments for different occasions. Their code is available at https://github.com/asFeng/kergnns, while being directly compatible to your library, i.e. using PyTorch and immanently compatible to your dataset formulations.

Best wishes, Tilman

Alternatives

No response

Additional context

No response

Guida contributor

Direzione di ricerca
Esamina il paper di KerGNN (https://arxiv.org/pdf/2201.00491.pdf) e l'implementazione di riferimento esistente su https://github.com/asFeng/kergnns. Studia come i layer GNN esistenti sono implementati nel modulo conv.nn di PyTorch Geometric, in particolare la classe base e il meccanismo di registrazione. Crea un nuovo modulo per KerGNN, assicurandone la compatibilità con le strutture dati della libreria e il paradigma di message passing, aggiungi test unitari e aggiorna la documentazione. Non ci sono PR aperti o assegnatari, quindi il lavoro è disponibile ma richiede un'attenta progettazione.
Tech stack
pythonpytorch
Dominio
machine learning
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
PythonPyTorchGraph Neural Networks
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
40