Skip to content

Commit bf57689

Browse files
committed
remove GAT from docstrings; not supported for static graphs.
1 parent 5ef2514 commit bf57689

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

flexynesis/modules.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)