/TaxoGAN

Code for the paper "Co-Embedding Network Nodes and Hierarchical Labels with Taxonomy Based Generative Adversarial Networks" ICDM 2020

Primary LanguagePython

Implementation of TaxoGAN, ICDM 2020.

Please cite the following work if you find the code useful.

@inproceedings{yang2020taxogan,
	Author = {Yang, Carl and Zhang, Jieyu and Han, Jiawei},
	Booktitle = {ICDM},
	Title = {Co-Embedding Network Nodes and Hierarchical Labels with Taxonomy Based Generative Adversarial Networks},
	Year = {2020}
}

Contact: Jieyu Zhang (jieyuz2@illinois.edu), Carl Yang (yangji9181@gmail.com)

Prerequisites

  • Python3
  • Pytorch 1.4

Training

python3 src/main.py --gpu 0 --dataset dblp --model TaxoGAN_V3 --task taxonomy --early_stop 0 --transform 1