fix bug in model_zoo

ispras · Nov 27, 2024 · a0a0803 · a0a0803
1 parent 0137cb8
commit a0a0803
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/experiments/attack_defense_test.py b/experiments/attack_defense_test.py
@@ -60,7 +60,7 @@ def test_attack_defense():
 
     # print(data.train_mask)
 
-    gnn = model_configs_zoo(dataset=dataset, model_name='gcn_gcn')
+    gnn = model_configs_zoo(dataset=dataset, model_name='gat_gcn_sage_gcn_gcn')
     # gnn = model_configs_zoo(dataset=dataset, model_name='gcn_gcn_lin')
     # gnn = model_configs_zoo(dataset=dataset, model_name='test_gnn')
     # gnn = model_configs_zoo(dataset=dataset, model_name='gin_gin_gin_lin_lin')

diff --git a/src/models_builder/models_zoo.py b/src/models_builder/models_zoo.py
@@ -183,7 +183,7 @@ def model_configs_zoo(
                             }
                         },
                         'activation': {
-                            'activation_name': 'ReLU',
+                            'activation_name': 'LeakyReLU',
                             'activation_kwargs': {
                                 "negative_slope": 0.01
                             },
@@ -196,11 +196,11 @@ def model_configs_zoo(
                             'layer_name': 'GCNConv',
                             'layer_kwargs': {
                                 'in_channels': 48,
-                                'out_channels': dataset.num_classes,
+                                'out_channels': 16,
                             },
                         },
                         'activation': {
-                            'activation_name': 'LeakyReLU',
+                            'activation_name': 'ReLU',
                             'activation_kwargs': None,
                         },
                     },
@@ -210,7 +210,7 @@ def model_configs_zoo(
                         'layer': {
                             'layer_name': 'SAGEConv',
                             'layer_kwargs': {
-                                'in_channels': dataset.num_node_features,
+                                'in_channels': 16,
                                 'out_channels': 16,
                             },
                         },
@@ -231,7 +231,7 @@ def model_configs_zoo(
                         'layer': {
                             'layer_name': 'GCNConv',
                             'layer_kwargs': {
-                                'in_channels': dataset.num_node_features,
+                                'in_channels': 16,
                                 'out_channels': 16,
                             },
                         },
@@ -246,8 +246,8 @@ def model_configs_zoo(
                         'layer': {
                             'layer_name': 'GCNConv',
                             'layer_kwargs': {
-                                'in_channels': dataset.num_node_features,
-                                'out_channels': 16,
+                                'in_channels': 16,
+                                'out_channels': dataset.num_classes,
                             },
                         },
                         'activation': {