2.3 Apply Message Passing On Part Of The Graph


If one only wants to update part of the nodes in the graph, the practice is to create a subgraph by providing the IDs for the nodes to include in the update, then call update_all() on the subgraph. For example:

nid = [0, 2, 3, 6, 7, 9]
sg = g.subgraph(nid)
sg.update_all(message_func, reduce_func, apply_node_func)

This is a common usage in mini-batch training. Check Chapter 6: Stochastic Training on Large Graphs for more detailed usages.