TaxoGAN
Code for the paper "Co-Embedding Network Nodes and Hierarchical Labels with Taxonomy Based Generative Adversarial Networks" ICDM 2020
Stars
5
Ecosystems:
Python
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 ([email protected]), Carl Yang ([email protected])
Prerequisites
- Python3
- Pytorch 1.4
Training
python3 src/main.py --gpu 0 --dataset dblp --model TaxoGAN_V3 --task taxonomy --early_stop 0 --transform 1