Pytorch Implementation of Adversarial Deep Network Embedding for Cross-Network Node Classification (ACDNE)
This is a pytorch implementation of the Adversarial Deep Network Embedding for Cross-Network Node Classification (ACDNE) model presented by Shen et. al (2020, https://arxiv.org/abs/2002.07366)
The official repository for ACDNE (Tensorflow) is available in https://github.com/shenxiaocam/ACDNE. Therefore, if you make advantage of the ACDNE model in your research, please cite the following:
Xiao Shen, Quanyu Dai, Fu-lai Chung, Wei Lu, and Kup-Sze Choi. Adversarial Deep Network Embedding for Cross-Network Node Classification. In Proceedings of AAAI Conference on Artificial Intelligence (AAAI), pages 2991-2999, 2020.
The code has been tested running under Python 3.8.8. The required packages are as follows:
• python == 3.8.8
• torch == 1.7.1
• numpy == 1.19.5
• scipy == 1.6.2
• sklearn == 0.24.1
input/ contains the 5 datasets used in our paper.
Each ".mat" file stores a network dataset, where
the variable "network" represents an adjacency matrix,
the variable "attrb" represents a node attribute matrix,
the variable "group" represents a node label matrix.
"ACDNE_model.py" is the implementation of the ACDNE model.
"ACDNE_test_Blog.py" is an example case of the cross-network node classification task from Blog1 to Blog2 networks.
"ACDNE_test_citation.py" is an example case of the cross-network node classification task from citationv1 to dblpv7 networks.