huihun/graphsage
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Small fix for type conversion.

Browse Source
This commit is contained in:
William Hamilton 2018-06-24 09:30:13 -04:00
parent 146406c2c0
commit d3105e5223
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graphsage/encoders.py
View File

@ -36,7 +36,7 @@ class Encoder(nn.Module):
nodes -- list of nodes
"""
neigh_feats = self.aggregator.forward(nodes, [self.adj_lists[node] for node in nodes],
neigh_feats = self.aggregator.forward(nodes, [self.adj_lists[int(node)] for node in nodes],
self.num_sample)
if not self.gcn:
if self.cuda: