A New Model for Predicting Node Type Based on Deep Learning

With the development of the Internet, a large number of data sets are generated, which contain valuable resources. Meanwhile, there are various graphical representations in real life, such as social networks, citation networks, and user networks. For user networks, there also exists rich information about entities except the network structure. Therefore, predicting the type of nodes in the network can help us quickly identify user type, citations type etc. In this paper, a new method based on deep learning is proposed to predict the class of node. Two public data sets are used as training sets. First, the node features are embedded to pre-train the neighbor’s neighborhood structure features, then the pre-trained data is used to input to the classification model, and the structural feature parameters are loaded. The final result shows that the prediction accuracy is increased by nearly 25% higher than the baseline model. The F1 scores of the model tested on the two data sets are 83.5% and 80.2%, respectively.