Computer Engineering & Science ›› 2021, Vol. 43 ›› Issue (02): 280-287.
Previous Articles Next Articles
CHEN Wen-qi1,WANG Ying1,3,WANG Xin2,3,WANG Hong-ji2
Received:
Revised:
Accepted:
Online:
Published:
Abstract: Node classification is widely used in social network and other network data. In order to study node classification, generative adversarial network (GAN) is used to obtain node representation, so as to obtain a good node classification effect. On this basis, a node classification-generative adversa- rial network (NC-GAN) model is proposed. This model uses GAN to conduct a binary game, considers the connectivity distribution in the network and the similarity between nodes to obtain the node representation that better fits the network, and then classifies the node representation to obtain a good classification effect. In order to verify the effect, the proposal is compared with DeepWalk, GraphGAN and other node representation model and graph convolutional network model in terms of link prediction and node classification. The model is only weaker than the GraphGAN model in link prediction, but it is better than other models in node classification.
Key words: generative adversarial network, multi-label, node classification
CHEN Wen-qi, WANG Ying, WANG Xin, WANG Hong-ji. Multi-label node classification based on generative adversarial network[J]. Computer Engineering & Science, 2021, 43(02): 280-287.
0 / / Recommend
Add to citation manager EndNote|Ris|BibTeX
URL: http://joces.nudt.edu.cn/EN/
http://joces.nudt.edu.cn/EN/Y2021/V43/I02/280