Skip to content

Commit

Permalink
Update run_GNE.py
Browse files Browse the repository at this point in the history
  • Loading branch information
kckishan authored Apr 11, 2018
1 parent ceb884e commit b55c2f3
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion run_GNE.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit b55c2f3

Please sign in to comment.