Update encoders.py

2017-12-27 16:23:07 -06:00 · 2017-12-27 16:23:07 -06:00 · 8e0d053da2
commit 8e0d053da2
parent b0bbc30be5
1 changed files with 0 additions and 1 deletions
--- a/graphsage/encoders.py
+++ b/graphsage/encoders.py
@ -35,7 +35,6 @@ class Encoder(nn.Module):
        Generates embeddings for a batch 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], 
                self.num_sample)