pyg-team/pytorch_geometric

Weighted aggregations `torch_geometric.nn.aggr.Aggregation`

Open

#6.867 geöffnet am 7. März 2023

Auf GitHub ansehen
 (10 Kommentare) (0 Reaktionen) (0 zugewiesene Personen)Python (3.514 Forks)batch import
1 - Priority P1featurehelp wantednn

Repository-Metriken

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

Beschreibung

🚀 The feature, motivation and pitch

The current API of torch_geometric.nn.aggr.Aggregation seems to be unnecessarily limited with respect to edge weights. At least some architectures introduce edge weights, with GCN being the most prominent one. For mean aggregation, it is simple to work around this with a custom message passing / convolution that multiplies features with edge weights. However, for most other weighted aggregations (e.g. median, quantiles, standard deviation, ...) it is not.

Alternatives

Add edge weights to aggregations API

Additional context

No response

Contributor Guide

Research-Richtung
Untersuche die aktuelle Implementierung der Aggregation Klassen in torch geometric.nn.aggr. Identifiziere, wo Kantengewichte nicht unterstützt werden. Betrachte vorhandene Workarounds im Repository (z. B. GCN Faltung). Schlage ein Design für die Hinzufügung von Kantengewichten zu Aggregationen wie median und quantiles vor. Diskutiere im Issue Thread.
Tech Stack
pythonpytorch
Domain
machine learning
Issue Type
Funktion
SchwierigkeitGeschätzte Implementierungsschwierigkeit für neue Contributors, von 1 für sehr kleine Änderungen bis 5 für Expertenarbeit.
3
Geschätzte ZeitEin grober Zeitrahmen, um zu recherchieren, zu implementieren, zu testen und einen Pull Request vorzubereiten.
1-2 Tage
AktivitätsstatusWie verfügbar das Issue gerade wirkt: frisch, aktiv, stale, blockiert oder wartet auf Maintainer-Input.
Aktiv
KlarheitWie klar das Issue die erwartete Änderung, Akzeptanzkriterien und den nächsten Schritt erklärt.
Klar
Voraussetzungen
PythonPyTorchGNN basics
EinsteigerfreundlichkeitEin Score von 1 bis 100, der schätzt, wie zugänglich dieses Issue für First-Time Contributors ist.
65