import torch.nn as nn from dgl.utils import expand_as_pair class SAGEConv(nn.Module): def __init__(self, in_feats, out_feats, aggregator_type, bias=True, norm=None, activation=None): super(SAGEConv, self).__init__() self._in_src_feats, self._in_dst_feats = expand_as_pair(in...
源节点特征feat_src和目标节点特征feat_dst需要根据图类型被指定。 用于指定图类型并将feat扩展为feat_src和feat_dst的函数是 :meth:~dgl.utils.expand_as_pair。 该函数的细节如下所示。 AI检测代码解析 def expand_as_pair(input_, g=None): if isinstance(input_, tuple): # 二分图的情况 return input_...
用于指定图类型并将feat扩展为feat_src和feat_dst的函数是expand_as_pair()。 该函数的细节如下所示。 1defexpand_as_pair(input_, g=None):2ifisinstance(input_, tuple):3#二分图的情况4returninput_5elifgisnotNoneandg.is_block:6#子图块的情况7ifisinstance(input_, Mapping):8input_dst ={9k: F...
from dgl import save_graphs, load_graphsimport dgl.function as fnimport torchimport dglimport torch.nn.functional as Ffrom dgl.nn.pytorch import GraphConv, SAGEConv, HeteroGraphConvfrom dgl.utils import expand_as_pairimport tqdmfrom collections import defaultdictimport torch as thimport dgl.nn as...
r} instead.'.format(valid_aggre_types,aggregator_type))self._in_src_feats,self._in_dst_feats=expand_as_pair(in_feats)self._out_feats=out_featsself._aggre_type=aggregator_typeself.norm=normself.feat_drop=nn.Dropout(feat_drop)self.activation=activationself.fc_self=nn.Linear(self._in_dst_...
feat_src, feat_dst = expand_as_pair(feat, graph) if self._norm == 'both': # clamp函数用于约束返回值到A和B之间,若value小于min,则返回min; # 若value大于max,则返回max,起到上下截断的作用。 degs = graph.out_degrees().float().clamp(min=1) # degs的-0.5次方norm = th.pow(degs, -0.5...
import numpy as npimport pandas as pdimport itertoolsimport osimport tqdmfrom dgl import save_graphs, load_graphsimport dgl.function as fnimport torchimport dglimport torch.nn.functional as Ffrom dgl.nn.pytorch import GraphConv, SAGEConv, HeteroGraphConvfrom dgl.utils import expand_as_pairimport...
load_graphsimport dgl.function as fnimport torchimport dglimport torch.nn.functional as Ffrom dgl.nn.pytorch import GraphConv, SAGEConv, HeteroGraphConvfrom dgl.utils import expand_as_pairimport tqdmfrom collections import defaultdictimport torch as thimport dgl.nn as dglnnfrom dgl.data.utils imp...
"""Torch Module for GraphSAGE layer"""# pylint: disable= no-member, arguments-differ, invalid-nameimporttorchfromtorchimportnnfromtorch.nnimportfunctionalasFfrom...importfunctionasfnfrom...utilsimportexpand_as_pair, check_eq_shapeclassSAGEConv(nn.Module):r""" Description...
load_graphsimport dgl.function as fnimport torchimport dglimport torch.nn.functional as Ffrom dgl.nn.pytorch import GraphConv, SAGEConv, HeteroGraphConvfrom dgl.utils import expand_as_pairimport tqdmfrom collections import defaultdictimport torch as thimport dgl.nn as dglnnfrom dgl.data.utils imp...