Skip to content
/ DBGAN Public

[CVPR2020] Tensorflow implementation for paper ''Distribution-induced Bidirectional Generative Adversarial Network for Graph Representation Learning''

Notifications You must be signed in to change notification settings

SsGood/DBGAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 

Repository files navigation

Distribution-induced Bidirectional GAN for Graph Representation Learning

This is a TensorFlow implementation of the Distribution-induced Bidirectional GAN (DBGAN) model as described in our paper. Some of the code is borrowed from T. N. Kipf, M. Welling, Variational Graph Auto-Encoders[https://github.com/tkipf/gae] and Hu. R, ARGA [https://github.com/Ruiqi-Hu/ARGA]

Introduction

This code contains two versions of the hyper-parameters. The first one is the implementation of node clustering task. The second one is the implementation of link prediction task.

Requirements

  • TensorFlow >= 1.12.0
  • python 3.6
  • networkx
  • scikit-learn
  • scipy
  • dppy==0.2.0
  • munkres

Run from

preset version:

python run.py

or modifying the network parameters and run

python run.py --hidden3 xxx --hidden2 xxx --learning_rate xxx ...

You can select the dataset in run.py

Data

If you want to use your own data, you have to provide

  • an N by N adjacency matrix (N is the number of nodes), and
  • an N by D feature matrix (D is the number of features per node) -- optional

Have a look at the load_data() function in input_data.py for an example.

In this example, we load citation network data (Cora, Citeseer or Pubmed). The original datasets can be found here: http://linqs.cs.umd.edu/projects/projects/lbc/ and here (in a different format): https://github.com/kimiyoung/planetoid

Cite

Please cite following papers if you use this code in your own work:

@inproceedings{zheng2020distribution,
  title={Distribution-induced bidirectional generative adversarial network for graph representation learning},
  author={Zheng, Shuai and Zhu, Zhenfeng and Zhang, Xingxing and Liu, Zhizhe and Cheng, Jian and Zhao, Yao},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={7224--7233},
  year={2020}
}

About

[CVPR2020] Tensorflow implementation for paper ''Distribution-induced Bidirectional Generative Adversarial Network for Graph Representation Learning''

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages