BVAT (ICML2019 workshop paper)
July 29, 2019 ยท View on GitHub
Code for "Batch Virtual Adversarial Training for Graph Convolutional Networks" which is based on the original implementation of GCN.
Requirements
- tensorflow (>0.12)
- networkx
Run the codes
We provide two adversarial training algorithms (SBVAT and OBVAT). Please refer to our paper for the details. Typically, you can run the algorithms by:
cd obvat
python train.py
Cite
Please cite our paper if you use this code in your own work:
@article{deng2019batch,
title={Batch Virtual Adversarial Training for Graph Convolutional Networks},
author={Deng, Zhijie and Dong, Yinpeng and Zhu, Jun},
journal={arXiv preprint arXiv:1902.09192},
year={2019}
}