(self, g)
| 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 = [] |
no test coverage detected