Skip to content

PyTorch implementation of some network embedding algorithms.

License

Notifications You must be signed in to change notification settings

so-link/NetworkEmbedding_Torch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 

Repository files navigation

NetworkEmbedding_Torch

Network embedding algorithms implementation with PyTorch.

This implementation runs with Anaconda3.

Paper

This repository contains source code of the following papers:

Homepage Augmentation by Predicting Links in Heterogeneous Networks. Jianming Lv, Jiajie Zhong, Weihang Chen, Qinzhe Xiao, Zhenguo Yang, and Qing Li. CIKM 2018. demo_ehwalk.py Dataset

ACE: Ant Colony Based Multi-Level Network Embedding for Hierarchical Graph Representation Learning. Jianming Lv, Jiajie Zhong, Jintao Liang, and Zhenguo Yang. IEEE Access. demo_ace.py

Quick Start

Setup the environment

$ cd src
$ pip install -r requirements.txt
$ python setup.py build_ext --inplace

Run demos

$ python demo_deepwalk.py
$ python demo_harp.py
$ python demo_ace.py

Docs

Under construction...

  • Overall introduction
  • Compile C++ extensions
  • Demos
  • Load datasets
  • Adapt your own coarsening algorithm to HARP or ACE
  • Adapt your own network embedding algorithm to HARP
  • Adapt your own network embedding algorithm to ACE

Benchmarks

Under construction...

Citations

Under construction...

Contact

If you have any question, please create an issue, or contact [email protected].

About

PyTorch implementation of some network embedding algorithms.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published