Commit dca2580b authored by shizhediao's avatar shizhediao Committed by Mufei Li
Browse files

fix minor error: GCNLayerSAGE->GraphSAGELayer (#1010)

fix minor error: GCNLayerSAGE->GraphSAGELayer
parent ead64de9
...@@ -78,15 +78,15 @@ class GraphSAGE(nn.Module): ...@@ -78,15 +78,15 @@ class GraphSAGE(nn.Module):
self.layers = nn.ModuleList() self.layers = nn.ModuleList()
# input layer # input layer
self.layers.append(GCNLayerSAGE(in_feats, n_hidden, activation=activation, self.layers.append(GraphSAGELayer(in_feats, n_hidden, activation=activation,
dropout=dropout, use_pp=use_pp, use_lynorm=True)) dropout=dropout, use_pp=use_pp, use_lynorm=True))
# hidden layers # hidden layers
for i in range(n_layers - 1): for i in range(n_layers - 1):
self.layers.append( self.layers.append(
GCNLayerSAGE(n_hidden, n_hidden, activation=activation, dropout=dropout, GraphSAGELayer(n_hidden, n_hidden, activation=activation, dropout=dropout,
use_pp=False, use_lynorm=True)) use_pp=False, use_lynorm=True))
# output layer # output layer
self.layers.append(GCNLayerSAGE(n_hidden, n_classes, activation=None, self.layers.append(GraphSAGELayer(n_hidden, n_classes, activation=None,
dropout=dropout, use_pp=False, use_lynorm=False)) dropout=dropout, use_pp=False, use_lynorm=False))
def forward(self, g): def forward(self, g):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment