how to build an heterogeneous graph network? about graph_nets HOT 3 CLOSED

Thanks for your message. The graph nets library does not currently support heterogeneous graphs in a single GraphsTuple.

The simple option would be to keep two GraphsTuples one with each of the types of edges, and with the same nodes. And then use the blocks.broadcasters/aggregator ops to build a model similar to the existing blocks.NodeUpdate and blocks.EdgeUpdate` that operates on both graphs in parallel.

Some options:

• I suspect the model may also work if you simply merge all edges into a single type, if all you are looking for informal reproducibility.
• At the edge update, you may want to use two separate MLPs for each edge type, by storing the edge type, then mask out the outputs corresponding to the wrong type for each MLP, and concatenate them side by side.
• Implementing the exact same baseline in an efficient way would require using the low level API of the GraphNets library as indicated above.

from graph_nets.

@alvarosg thanks for your help.I will try to implement your advise approach.

Besides, may i ask you a question? When will you publish the code of the paper?

from graph_nets.

Code is now available here https://github.com/deepmind/deepmind-research/tree/master/meshgraphnets :)

from graph_nets.