GCN 自定义数据集
流程
使用的是DGL框架
为什么使用DGL 不使用pyG 因为DGL大图支持更好
定义自己的图矩阵
#An highlighted block
import dgl
import numpy as np
# S D source destination 类型 list【int】
def build_graph(S,D):
# 另一个用于目标端点。
src = np.array(S)
dst = np.array(D)
# 在DGL中,边是有方向性的;让他们双向。
# u = np.concatenate([src, dst])
# v = np.concatenate([dst, src])
# 构造一个DGLGraph
# return dgl.DGLGraph((u, v))
return dgl.DGLGraph((src, dst))
# 请自己定义
S,D = 自行填写
G = build_graph(S,D)
print(