network-representation-learning

Four network representation learning algorithms have been reimplementationed

• Deepwalk, assess on tencent weibo dataset, AUC: 0.7548
• LINE, assess on tencent weibo dataset, AUC: 0.7608
• Node2Vec, assess on tencent weibo dataset, AUC: 0.7553
• GraRep, assess on cora dataset, prediction accuracy: 0.805

Usage

Go to the source directory, use following command to run:

python3 deepwalk_for_tencent.py [or line_for_tencent.py, node2vec_for_tencent.py, grarep_for_cora.py]

Requirements

• numpy
• scipy
• networksx
• gensim
• pytorch
• scikit-learn