pyg-team/pytorch_geometric

Weighted aggregations `torch_geometric.nn.aggr.Aggregation`

開放

#6,867 建立於 2023年3月7日

 (10 則留言) (0 個反應) (0 位負責人)Python (3,514 個分叉)batch import
1 - Priority P1featurehelp wantednn

倉庫指標

星標
 (19,985 顆星)
PR 合併指標
 (PR 指標待抓取)

描述

🚀 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

貢獻者指南