pyg-team/pytorch_geometric
GitHub ã§èŠãWeighted aggregations `torch_geometric.nn.aggr.Aggregation`
Open
#6,867 opened on 2023幎3æ7æ¥
1 - Priority P1featurehelp wantednn
Repository metrics
- Stars
- Â (19,985 stars)
- PR merge metrics
-  (å¹³åããŒãž 35d 1h) (30d 㧠14 merged PRs)
説æ
ð 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