使用DGL都早异构图
参考:https://docs.dgl.ai/guide_cn/graph-heterogeneous.html
import dgl
import torch
Using backend: pytorch
使用DGL创建异构图
下面一个异构图示例。该图具有两种类型的节点(“用户”和”游戏”)和两种类型的边(“关注”和”玩”)。
我们将按照这个例子来创建一个异构图。
graph_data = {
('user', 'follows', 'user') : (torch.tensor([0]), torch.tensor([1])),
('user', 'plays', 'game') : (torch.tensor([0, 0, 1, 1]), torch.tensor([0, 1, 1, 2]))
}
g = dgl.heterograph(graph_data)
print(g)
Graph(num_nodes={'game': 3, 'user': 2},
num_edg