Update VGCL.py

yimutianyang · May 31, 2023 · 403bd31 · 403bd31
1 parent e2262de
commit 403bd31
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/models/VGCL.py b/models/VGCL.py
@@ -40,8 +40,8 @@ def graph_encoder(self):
         ### reparameterization
         noise1 = tf.random_normal(std.shape)
         noise2 = tf.random_normal(std.shape)
-        noised_emb1 = mean + 0.01 * std * noise1
-        noised_emb2 = mean + 0.01 * std * noise2
+        noised_emb1 = mean + std * noise1
+        noised_emb2 = mean + std * noise2
         return noised_emb1, noised_emb2, mean, std
 
 
@@ -155,4 +155,4 @@ def compute_cl_loss_cluster(self):
         log_probs_item = tf.math.divide_no_nan(log_probs_item, num_item_pos_per_cow)
         cl_loss_item = -tf.reduce_mean(tf.math.log(log_probs_item))
         cl_loss = self.alpha * (cl_loss_user + cl_loss_item)
-        return cl_loss
+        return cl_loss