[發明專利]為二分圖中的節點分配嵌入向量的方法以及裝置

在審

• 申請號: 201911310875.0
• 申請日: 2019-12-18
• 公開（公告）號: CN111176838A
• 公開（公告）日: 2020-05-19
• 發明（設計）人: 蘇煒躍;馮仕堃;朱志凡;何徑舟
• 申請（專利權）人: 北京百度網訊科技有限公司
• 主分類號: G06F9/50
• 分類號: G06F9/50
• 代理公司: 北京鴻德海業知識產權代理事務所(普通合伙) 11412
• 代理人: 田宏賓

1.一種為二分圖中的節點分配嵌入向量的方法，其特征在于，包括：

獲取待處理二分圖；

確定所述待處理二分圖中的少數類節點以及多數類節點；

為各少數類節點分配對應各自的嵌入向量，為各多數類節點分配一個共享的嵌入向量，輸出所述待處理二分圖的向量分配結果。

2.根據權利要求1所述的方法，其特征在于，所述確定所述待處理二分圖中的少數類節點以及多數類節點包括：

分別獲取所述待處理二分圖中兩個節點集合所包含節點的數量；

將節點數量較少的節點集合中的各節點作為少數類節點，將另一節點集合中的各節點作為多數類節點。

3.根據權利要求1所述的方法，其特征在于，所述方法還包括：

確定與各多數類節點具有連接關系的鄰居節點；

將對應同一個多數類節點的各鄰居節點對應的嵌入向量進行拼接，將拼接結果作為對應該多數類節點的嵌入向量。

4.根據權利要求3所述的方法，其特征在于，所述將對應同一個多數類節點的各鄰居節點對應的嵌入向量進行拼接包括：

獲取各鄰居節點對應的嵌入向量；

將各鄰居節點對應的嵌入向量與所述共享的嵌入向量進行拼接。

5.一種為二分圖中的節點分配嵌入向量的裝置，其特征在于，包括：

獲取單元，用于獲取待處理二分圖；

確定單元，用于確定所述待處理二分圖中的少數類節點以及多數類節點；

分配單元，用于為各少數類節點分配對應各自的嵌入向量，為各多數類節點分配一個共享的嵌入向量，輸出所述待處理二分圖的向量分配結果。

6.根據權利要求5所述的裝置，其特征在于，所述確定單元在確定所述待處理二分圖中的少數類節點以及多數類節點時，具體執行：

分別獲取所述待處理二分圖中兩個節點集合所包含節點的數量；

將節點數量較少的節點集合中的各節點作為少數類節點，將另一節點集合中的各節點作為多數類節點。

7.根據權利要求5所述的裝置，其特征在于，所述分配單元還執行：

確定與各多數類節點具有連接關系的鄰居節點；

將對應同一個多數類節點的各鄰居節點對應的嵌入向量進行拼接，將拼接結果作為對應該多數類節點的嵌入向量。

8.根據權利要求7所述的裝置，其特征在于，所述分配單元在將對應同一個多數類節點的各鄰居節點對應的嵌入向量進行拼接時，具體執行：

獲取各鄰居節點對應的嵌入向量；

將各鄰居節點對應的嵌入向量與所述共享的嵌入向量進行拼接。

9.一種電子設備，其特征在于，包括：

至少一個處理器；以及

與所述至少一個處理器通信連接的存儲器；其中，

所述存儲器存儲有可被所述至少一個處理器執行的指令，所述指令被所述至少一個處理器執行，以使所述至少一個處理器能夠執行權利要求1-4中任一項所述的方法。

10.一種存儲有計算機指令的非瞬時計算機可讀存儲介質，其特征在于，所述計算機指令用于使所述計算機執行權利要求1-4中任一項所述的方法。