@@ -175,8 +175,7 @@ class flexGCN(nn.Module):
175175 dropout_rate (float, optional): The dropout probability used for regularization. Defaults to 0.2.
176176 conv (str, optional): Type of convolution layer to use. Supported types
177177 include 'GCN' for Graph Convolution Network,
178- 'GAT' for Graph Attention Network, 'SAGE' for
179- GraphSAGE, and 'GC' for generic Graph Convolution.
178+ 'SAGE' for GraphSAGE, and 'GC' for generic Graph Convolution.
180179 Defaults to 'GC'.
181180 act (str, optional): Type of activation function to use. Supported types
182181 include 'relu', 'sigmoid', 'leakyrelu', 'tanh',
@@ -187,7 +186,7 @@ class flexGCN(nn.Module):
187186
188187 Example:
189188 >>> model = flexGCN(node_count=100, node_feature_count=5, node_embedding_dim=64, output_dim=10,
190- num_convs=3, dropout_rate=0.3, conv='GAT ', act='relu')
189+ num_convs=3, dropout_rate=0.3, conv='SAGE ', act='relu')
191190 >>> output = model(input_features, edge_index)
192191 # Where `input_features` is a tensor of shape (batch_size, num_nodes, node_feature_count)
193192 # and `edge_index` is a list of edges in the COO format (2, num_edges).
0 commit comments