自定义数据跑GCN
GCN 自定义数据集流程定义自己的图矩阵定义GCN模型设定特征矩阵 及 标签训练流程使用的是DGL框架为什么使用DGL 不使用pyG 因为DGL大图支持更好定义自己的图矩阵#An highlighted blockimport dglimport numpy as np# S D source destination 类型 list【int】def build_graph(S,D): # 另一个用于目标端点。 src = np.array(S) dst = n
原创
2020-09-22 02:05:23 ·
1225 阅读 ·
4 评论