/network-representation-learning

network-representation-learning: DeepWalk, LINE, Node2Vec, GraRep

Primary LanguagePython

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