From 53bfbabcd73e01084790718105f4734e03eb3253 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Fri, 24 Feb 2023 21:32:02 +0000 Subject: [PATCH] gnn_4 --- models/gnn.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/models/gnn.py b/models/gnn.py index 7414009..0e55b7b 100644 --- a/models/gnn.py +++ b/models/gnn.py @@ -16,9 +16,9 @@ class Smoother(tf.keras.Model): def __init__(self, ): super().__init__() self.a = sp_matrix_to_sp_tensor(sio.loadmat("custom_utils/ssmpl0.mat")['L'].tocoo()) - self.conv1 = GCNConv(64, activation='tanh', use_bias=False) - self.conv2 = GCNConv(64, activation='tanh', use_bias=False) - self.conv3 = GCNConv(64, activation='tanh', use_bias=False) + self.conv1 = GCNConv(3, activation='tanh', use_bias=False) + self.conv2 = GCNConv(3, activation='tanh', use_bias=False) + self.conv3 = GCNConv(3, activation='tanh', use_bias=False) # self.conv4 = GCNConv(64, activation='tanh', use_bias=False) # self.conv5 = GCNConv(64, activation='tanh', use_bias=False) self.conv6 = GCNConv(3, activation='tanh', use_bias=False)