pyg-team/pytorch_geometric

Support `AttentionalAggregation` for `HeteroConv`

Open

#7 414 ouverte le 23 mai 2023

Voir sur GitHub
 (1 commentaire) (0 réactions) (0 assignés)Python (3 514 forks)batch import
1 - Priority P1featurehelp wanted

Métriques du dépôt

Stars
 (19 985 stars)
Métriques de merge PR
 (Merge moyen 35j 1h) (14 PRs mergées en 30 j)

Description

🚀 The feature, motivation and pitch

use pyg == 2.3.0
conv = HeteroConv({
    edge_type: GraphConv(-1, second_channels) for edge_type in metadata[1]
}, aggr='attentional')

In pyg 2.3.0, can not use aggr.AttentionalAggregation to aggregation node feature from different edge type in above code, right ? (may i thought is wrong...), so can you give me some advice on how to integrate node features from different edges in pyg ?

Alternatives

No response

Additional context

No response

Guide contributeur