PREDICTION METHOD AND SYSTEM BASED ON HETEROGENEOUS GRAPH NEURAL NETWORK MODEL

摘要

A prediction method and a system based on a heterogeneous graph neural network model. The method comprises: acquiring heterogeneous graph data related to prediction content, wherein the heterogeneous graph data comprises a node to be subjected to prediction, neighbor nodes of the node to be subjected to prediction, and paths connected between the node to be subjected to prediction and the neighbor nodes, wherein the paths comprise at least one type (302); grouping the neighbor nodes on the basis of the type of the paths, such that the types of paths of the neighbor nodes of the same group are the same (304); and inputting the node to be subjected to prediction, the grouped neighbor nodes, and the paths between the nodes into a trained heterogeneous graph neural network model, so as to obtain a representation vector of the node to be subjected to prediction, and then inputting the representation vector into a trained prediction model for prediction (306).