MCPcopy
hub / github.com/dmlc/dgl / gcn

Method gcn

python/dgl/transforms/module.py:1844–1863  ·  view source on GitHub ↗
(self, g)

Source from the content-addressed store, hash-verified

1842 return feat_list
1843
1844 def gcn(self, g):
1845 feat_list = []
1846 with g.local_scope():
1847 if self.eweight_name is None:
1848 eweight_name = "w"
1849 if eweight_name in g.edata:
1850 g.edata.pop(eweight_name)
1851 else:
1852 eweight_name = self.eweight_name
1853
1854 transform = GCNNorm(eweight_name=eweight_name)
1855 transform(g)
1856
1857 for _ in range(self.k):
1858 g.update_all(
1859 fn.u_mul_e(self.in_feat_name, eweight_name, "m"),
1860 fn.sum("m", self.in_feat_name),
1861 )
1862 feat_list.append(g.ndata[self.in_feat_name])
1863 return feat_list
1864
1865 def ppr(self, g):
1866 feat_list = []

Callers 1

forwardMethod · 0.80

Calls 5

GCNNormClass · 0.85
transformFunction · 0.85
local_scopeMethod · 0.80
update_allMethod · 0.80
appendMethod · 0.80

Tested by

no test coverage detected