test sample with Batch and sage with concat

xiang526 · Mar 30, 2020 · 3f147fe · 3f147fe
1 parent e97ef56
commit 3f147fe
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gcn.py b/gcn.py
@@ -143,11 +143,13 @@ def __init__(self, in_channels, out_channels, gcn_type, gcn_partition=None):
         else:
             if gcn_type == 'gat':
                 self.adj_available = False
-            if gcn_type in ['normal', 'cheb', 'graph']:
+            if gcn_type in ['normal', 'cheb']:
                 self.batch_training = True
                 self.kwargs['node_dim'] = 1
             if gcn_type == 'cheb':
                 self.kwargs['K'] = 3
+            if gcn_type == 'sage':
+                self.kwargs['concat'] = True
 
             GCNCell = {'normal':PyG.GCNConv, 
                         'cheb':PyG.ChebConv,