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

Update encoders.py

Browse Source
This commit is contained in:
William L Hamilton 2017-12-27 16:23:07 -06:00 committed by GitHub
parent b0bbc30be5
commit 8e0d053da2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
graphsage/encoders.py
View File

@ -35,7 +35,6 @@ class Encoder(nn.Module):
Generates embeddings for a batch of nodes. Generates embeddings for a batch of nodes.
nodes -- list of nodes nodes -- list of nodes
mode -- string desiginating the mode of the nodes
""" """
neigh_feats = self.aggregator.forward(nodes, [self.adj_lists[node] for node in nodes], neigh_feats = self.aggregator.forward(nodes, [self.adj_lists[node] for node in nodes],
self.num_sample) self.num_sample)