利用小批量训练的方法在子图中进行消息传递

如果用户只想更新图中的部分节点,可以先通过想要囊括的节点编号创建一个子图, 然后在子图上调用 update_all() 方法。例如:

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

同时也可以在大图中使用随机(批次)训练。第6章:在大图上的随机(批次)训练

你可能感兴趣的:(GCN,图卷积网络,图神经网络,python,GNN)