search

tensorflow / tfjs

A WebGL accelerated JavaScript library for training and deploying ML models.
https://js.tensorflow.org
Apache License 2.0
18.36k stars 1.92k forks source link

Graph Neural Network (GNN) support in TensorFlow.js #5975

Open jasonmayes opened 2 years ago

jasonmayes commented 2 years ago

Currently TensorFlow.js does not support the new GNN (Graph based neural networks) as detailed here. A growing number of members of the community are keen to see this feature implemented in TensorFlow.js to keep up to date with the latest research.

Will this change the current api? How? This will extend the current existing API to support new features related to GNNs

Who will benefit with this feature? All users of TensorFlow.js using to create GNN based models.

Any Other info. Please also see this forum discussion: https://discuss.tensorflow.org/t/gnn-for-tensorflow-js/6747

grmatthews commented 2 years ago

Am willing to support this as a contributor. Contact me if/when there's enough support to kick start this.

jasonmayes commented 2 years ago

Any updates on this one if feasible for the future to look into?

grmatthews commented 2 years ago

Can I request that we propose to look into building an RGCN (Relational GCN), which would have multiple node types, multiple edge types, and where the edges and nodes each can have features? I am mentioning this since 'GCN' could mean what I've described, or a subset, e.g. homogenous nodes, no edge features, etc. And I'm calling it a RGCN since that's what Zak Jost calls it.