[发明专利]为二分图中的节点分配嵌入向量的方法以及装置

权利要求书

1.一种为二分图中的节点分配嵌入向量的方法，其特征在于，包括：

获取待处理二分图；

确定所述待处理二分图中的少数类节点以及多数类节点；

为各少数类节点分配对应各自的嵌入向量，为各多数类节点分配一个共享的嵌入向量，输出所述待处理二分图的向量分配结果。

2.根据权利要求1所述的方法，其特征在于，所述确定所述待处理二分图中的少数类节点以及多数类节点包括：

分别获取所述待处理二分图中两个节点集合所包含节点的数量；

将节点数量较少的节点集合中的各节点作为少数类节点，将另一节点集合中的各节点作为多数类节点。

3.根据权利要求1所述的方法，其特征在于，所述方法还包括：

确定与各多数类节点具有连接关系的邻居节点；

将对应同一个多数类节点的各邻居节点对应的嵌入向量进行拼接，将拼接结果作为对应该多数类节点的嵌入向量。

4.根据权利要求3所述的方法，其特征在于，所述将对应同一个多数类节点的各邻居节点对应的嵌入向量进行拼接包括：

获取各邻居节点对应的嵌入向量；

将各邻居节点对应的嵌入向量与所述共享的嵌入向量进行拼接。

5.一种为二分图中的节点分配嵌入向量的装置，其特征在于，包括：

获取单元，用于获取待处理二分图；

确定单元，用于确定所述待处理二分图中的少数类节点以及多数类节点；

分配单元，用于为各少数类节点分配对应各自的嵌入向量，为各多数类节点分配一个共享的嵌入向量，输出所述待处理二分图的向量分配结果。

6.根据权利要求5所述的装置，其特征在于，所述确定单元在确定所述待处理二分图中的少数类节点以及多数类节点时，具体执行：

分别获取所述待处理二分图中两个节点集合所包含节点的数量；

将节点数量较少的节点集合中的各节点作为少数类节点，将另一节点集合中的各节点作为多数类节点。

7.根据权利要求5所述的装置，其特征在于，所述分配单元还执行：

确定与各多数类节点具有连接关系的邻居节点；

将对应同一个多数类节点的各邻居节点对应的嵌入向量进行拼接，将拼接结果作为对应该多数类节点的嵌入向量。

8.根据权利要求7所述的装置，其特征在于，所述分配单元在将对应同一个多数类节点的各邻居节点对应的嵌入向量进行拼接时，具体执行：

获取各邻居节点对应的嵌入向量；

将各邻居节点对应的嵌入向量与所述共享的嵌入向量进行拼接。

9.一种电子设备，其特征在于，包括：

至少一个处理器；以及

与所述至少一个处理器通信连接的存储器；其中，

所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行权利要求1-4中任一项所述的方法。

10.一种存储有计算机指令的非瞬时计算机可读存储介质，其特征在于，所述计算机指令用于使所述计算机执行权利要求1-4中任一项所述的方法。