search

pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch
https://pyg.org
MIT License
21.23k stars 3.64k forks source link

Multipartite graphs support #5003

Open YacineBenameur opened 2 years ago

YacineBenameur commented 2 years ago

🚀 The feature, motivation and pitch

Hi,

Similar to the issue on bipartite graphs bipartite graphs, i would like to know if it's possible (and how :) ) to create Message Passing Network for multi-partite graphs. A highly desired attribute of this network is the ability to define different node features for each separate set of nodes.

Thanks in advance :)

Alternatives

No response

Additional context

No response

rusty1s commented 2 years ago

Please take a look at https://pytorch-geometric.readthedocs.io/en/latest/notes/heterogeneous.html

YacineBenameur commented 2 years ago

Hi @rusty1s , this is actually what I was looking for.

Many thanks :)