/NetMF

Network Embedding as Matrix Factorization: Unifying DeepWalk, LINE, PTE, and node2vec

Primary LanguagePythonMIT LicenseMIT

NetMF

This is a Python implementation of NetMF for the task of network embedding learning, as described in our paper:

Network Embedding as Matrix Factorization: Unifying DeepWalk, LINE, PTE, and node2vec

Data Sets

BlogCatalog

Protein-Protein Interaction Source Preprocessed

Wikipedia Source Preprocessed

Flickr

Cite

Please cite our paper if you use this code in your own work:

@inproceedings{qiu2018network,
  title={Network Embedding as Matrix Factorization: Unifying DeepWalk, LINE, PTE, and node2vec},
  author={Qiu, Jiezhong and Dong, Yuxiao and Ma, Hao and Li, Jian and Wang, Kuansan and Tang, Jie},
  booktitle={Proceedings of the Eleventh ACM International Conference on Web Search and Data Mining},
  pages={459--467},
  year={2018},
  organization={ACM}
}