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

说明书

本申请公开了一种为二分图中的节点分配嵌入向量的方法、装置、电子设备以及计算机可读存储介质，涉及图处理技术领域。本申请在为二分图中的节点分配嵌入向量时的实现方案为：获取待处理二分图；确定所述待处理二分图中的少数类节点以及多数类节点；为各少数类节点分配对应各自的嵌入向量，为各多数类节点分配一个共享的嵌入向量，输出所述待处理二分图的向量分配结果。本申请能够提升嵌入向量的分配效率，降低嵌入向量所占用的内存资源。

技术领域

本申请涉及数据处理技术领域，尤其涉及图处理技术领域中的一种为二分图中的节点分配嵌入向量的方法、装置、电子设备以及计算机可读存储介质。

背景技术

现有技术中，在为二分图中的节点分配嵌入向量时，通常是对二分图中的各节点分配对应各自的嵌入向量。因此，当所要处理的二分图的数量非常多时，分配各二分图中的全部节点的嵌入向量需要耗费较多大量时间，且对各节点的嵌入向量进行保存时需要占用较大的内存资源。

发明内容

本申请为解决技术问题所采用的技术方案是提供一种为二分图中的节点分配嵌入向量的方法、装置、电子设备以及计算机可读介质，所述方法包括：获取待处理二分图；确定所述待处理二分图中的少数类节点以及多数类节点；为各少数类节点分配对应各自的嵌入向量，为各多数类节点分配一个共享的嵌入向量，输出所述待处理二分图的向量分配结果。本申请能够提升嵌入向量的分配效率，降低嵌入向量所占用的内存资源。

根据本申请一优选实施例，所述确定所述待处理二分图中的少数类节点以及多数类节点包括：分别获取所述待处理二分图中两个节点集合所包含节点的数量；将节点数量较少的节点集合中的各节点作为少数类节点，将另一节点集合中的各节点作为多数类节点。本步骤能够提升节点划分的准确性。

根据本申请一优选实施例，所述方法还包括：确定与各多数类节点具有连接关系的邻居节点；将对应同一个多数类节点的各邻居节点对应的嵌入向量进行拼接，将拼接结果作为对应该多数类节点的嵌入向量。本步骤在无需为多数类节点预先分配嵌入向量的情况下，能够利用已知的节点之间的连接关系和少数类节点的嵌入向量来获取多数类节点的嵌入向量。

根据本申请一优选实施例，所述将对应同一个多数类节点的各邻居节点对应的嵌入向量进行拼接包括：获取各邻居节点对应的嵌入向量；将各邻居节点对应的嵌入向量与所述共享的嵌入向量进行拼接。

本申请为解决技术问题所采用的技术方案是提供一种为二分图中的节点分配嵌入向量的装置，包括：获取单元，用于获取待处理二分图；确定单元，用于确定所述待处理二分图中的少数类节点以及多数类节点；分配单元，用于为各少数类节点分配对应各自的嵌入向量，为各多数类节点分配一个共享的嵌入向量，输出所述待处理二分图的向量分配结果。

根据本申请一优选实施例，所述确定单元在确定所述待处理二分图中的少数类节点以及多数类节点时，具体执行：分别获取所述待处理二分图中两个节点集合所包含节点的数量；将节点数量较少的节点集合中的各节点作为少数类节点，将另一节点集合中的各节点作为多数类节点。

根据本申请一优选实施例，所述分配单元还执行：确定与各多数类节点具有连接关系的邻居节点；将对应同一个多数类节点的各邻居节点对应的嵌入向量进行拼接，将拼接结果作为对应该多数类节点的嵌入向量。

根据本申请一优选实施例，所述分配单元在将对应同一个多数类节点的各邻居节点对应的嵌入向量进行拼接时，具体执行：获取各邻居节点对应的嵌入向量；将各邻居节点对应的嵌入向量与所述共享的嵌入向量进行拼接。

上述申请中的一个实施例具有如下优点或有益效果：本申请能够提升嵌入向量的分配效率，降低为节点所分配的嵌入向量占用的内存资源。因为采用了只对二分图中的部分节点分配各自所对应的嵌入向量的方式，所以克服了现有技术中对全部节点分配各自所对应的嵌入向量所导致的分配效率低、内存资源占用大的技术问题，达到提升嵌入向量的分配效率、降低嵌入向量所占用的内存资源的技术效果。