![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
DGL
cici_iii
keep coding
展开
-
DGL实现同构/异构图卷积模型
同构图卷积from time import timeimport numpy as npimport dglimport torchimport torch.nn as nnimport torch.nn.functional as F# 模型class TwoLayerModel(nn.Module): def __init__(self): super().__init__() # gcn self.conv1 = dgl.nn原创 2021-10-21 15:33:34 · 1213 阅读 · 2 评论 -
DGL_子图
用途一:数据集太大,无法画图,取子图看看是有向图/无向图import dglimport matplotlib.pyplot as pltimport networkx as nxG = dgl.DGLGraph()G.add_nodes(5)# G.add_edges([0, 1, 2, 3, 4], [1, 2, 3, 4, 0]) # 有向图G.add_edges([0, 1, 2, 3, 4, 1, 2, 3, 4, 0], [1, 2, 3, 4, 0, 0, 1, 2, 3,原创 2020-07-24 15:44:20 · 737 阅读 · 0 评论 -
DGL_图的打印
首先要安装 networkximport matplotlib.pyplot as pltimport networkx as nximport dglimport numpy as npdef build_karate_club_graph(): src = np.array([1, 2, 2, 3, 3]) dst = np.array([0, 0, 1, 0, 1]) u = np.concatenate([src, dst]) v = np.concaten原创 2020-07-23 16:15:50 · 449 阅读 · 0 评论 -
DGL_图的创建、保存、加载
import dglimport torch as thfrom dgl.data.utils import save_graphsg1 = dgl.DGLGraph()g1.add_nodes(3)g1.add_edges([0, 0, 0, 1, 1, 2], [0, 1, 2, 1, 2, 2])g1.ndata["x"] = th.ones(3, 5) # 3个节点的embeddingg1.edata['y'] = th.zeros(6, 5) # 6条边的embedding原创 2020-07-23 15:31:13 · 2984 阅读 · 3 评论