公开(公告)号：US20210201198A1

公开(公告)日：2021-07-01

申请号：US16945183

申请日：2020-07-31

Applicant: BEIJING BAIDU NETCOM SCIENCE AND TECHNOLOGY CO., LTD.

Inventor： Weibin LI ,  Zhifan ZHU ,  Weiyue SU ,  Jingzhou HE ,  Shikun FENG ,  Yuhui CAO ,  Xuyi CHEN ,  Danxiang ZHU

IPC: G06N20/00 ,  G06F16/901

Abstract: A method for generating node representations in a heterogeneous graph, an electronic device, and a non-transitory computer-readable storage medium, and relates to the field of machine learning technologies. The method includes: acquiring a heterogeneous graph; inputting the heterogeneous graph into a heterogeneous graph learning model to generate a node representation of each node in the heterogeneous graph, in which the heterogeneous graph learning model generates the node representation of each node by actions of: segmenting the heterogeneous graph into a plurality of subgraphs, in which each subgraph includes nodes of two types and an edge of one type between the nodes of two types; and generating the node representation of each node according to the plurality of subgraphs.