Skip to content

Latest commit

 

History

History
 
 

ges

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
README.md
README.md
 
 
gpu_run.sh
gpu_run.sh
 
 
gpu_train.py
gpu_train.py
 
 
model.py
model.py
 
 
mp_reader.py
mp_reader.py
 
 
reader.py
reader.py
 
 

README.md

GES: Graph Embedding with Side Information

Graph Embedding with Side Information is an algorithmic framework for representational learning on graphs. Given any graph, it can learn continuous feature representations for the nodes, which can then be used for various downstream machine learning tasks. Based on PGL, we reproduce ges algorithms.

Datasets

The datasets contain two networks: BlogCatalog.

Dependencies

  • paddlepaddle>=1.6
  • pgl>=1.0.0

How to run

For examples, train ges on cora dataset.

# train deepwalk in distributed mode.
sh gpu_run.sh

Hyperparameters

  • dataset: The citation dataset "BlogCatalog".
  • hidden_size: Hidden size of the embedding.
  • lr: Learning rate.
  • neg_num: Number of negative samples.
  • epoch: Number of training epoch.