From b55c2f392822a89984bd7f4c6f0cc7d12e6f4294 Mon Sep 17 00:00:00 2001
From: "Kishan K.C" <kckishan@users.noreply.github.com>
Date: Wed, 11 Apr 2018 16:13:09 -0400
Subject: [PATCH] Update run_GNE.py

---
 run_GNE.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/run_GNE.py b/run_GNE.py
index e4b3b45..4838ce7 100755
--- a/run_GNE.py
+++ b/run_GNE.py
@@ -7,13 +7,22 @@
 
 ################################# Define parameters to train GNE model #######################################
 parameters = {}
+# Dimension of topological structure embeddings
 parameters['id_embedding_size'] = 128
+# Dimension of expression data embeddings
 parameters['attr_embedding_size'] = 128
+# Dimension of final representation after transformation of concatenated topological properties and expression data representation
+parameters['representation_size'] = 128
+# Importance of gene expression relative to topological properties
 parameters['alpha'] = 1
+
+# Number of negative samples for Negative Sampling
 parameters['n_neg_samples'] = 10
+# Number of epochs to run the model
 parameters['epoch'] = 20
-parameters['representation_size'] = 128
+# Number of sample to consider in the batch
 parameters['batch_size'] = 256
+# Learning rate
 parameters['learning_rate'] = 0.005
 
 print(parameters)