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

在審

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

本申請公開了一種為二分圖中的節點分配嵌入向量的方法、裝置、電子設備以及計算機可讀存儲介質，涉及圖處理技術領域。本申請在為二分圖中的節點分配嵌入向量時的實現方案為：獲取待處理二分圖；確定所述待處理二分圖中的少數類節點以及多數類節點；為各少數類節點分配對應各自的嵌入向量，為各多數類節點分配一個共享的嵌入向量，輸出所述待處理二分圖的向量分配結果。本申請能夠提升嵌入向量的分配效率，降低嵌入向量所占用的內存資源。

技術領域

本申請涉及數據處理技術領域，尤其涉及圖處理技術領域中的一種為二分圖中的節點分配嵌入向量的方法、裝置、電子設備以及計算機可讀存儲介質。

背景技術

現有技術中，在為二分圖中的節點分配嵌入向量時，通常是對二分圖中的各節點分配對應各自的嵌入向量。因此，當所要處理的二分圖的數量非常多時，分配各二分圖中的全部節點的嵌入向量需要耗費較多大量時間，且對各節點的嵌入向量進行保存時需要占用較大的內存資源。

發明內容

本申請為解決技術問題所采用的技術方案是提供一種為二分圖中的節點分配嵌入向量的方法、裝置、電子設備以及計算機可讀介質，所述方法包括：獲取待處理二分圖；確定所述待處理二分圖中的少數類節點以及多數類節點；為各少數類節點分配對應各自的嵌入向量，為各多數類節點分配一個共享的嵌入向量，輸出所述待處理二分圖的向量分配結果。本申請能夠提升嵌入向量的分配效率，降低嵌入向量所占用的內存資源。

根據本申請一優選實施例，所述確定所述待處理二分圖中的少數類節點以及多數類節點包括：分別獲取所述待處理二分圖中兩個節點集合所包含節點的數量；將節點數量較少的節點集合中的各節點作為少數類節點，將另一節點集合中的各節點作為多數類節點。本步驟能夠提升節點劃分的準確性。

根據本申請一優選實施例，所述方法還包括：確定與各多數類節點具有連接關系的鄰居節點；將對應同一個多數類節點的各鄰居節點對應的嵌入向量進行拼接，將拼接結果作為對應該多數類節點的嵌入向量。本步驟在無需為多數類節點預先分配嵌入向量的情況下，能夠利用已知的節點之間的連接關系和少數類節點的嵌入向量來獲取多數類節點的嵌入向量。

根據本申請一優選實施例，所述將對應同一個多數類節點的各鄰居節點對應的嵌入向量進行拼接包括：獲取各鄰居節點對應的嵌入向量；將各鄰居節點對應的嵌入向量與所述共享的嵌入向量進行拼接。

本申請為解決技術問題所采用的技術方案是提供一種為二分圖中的節點分配嵌入向量的裝置，包括：獲取單元，用于獲取待處理二分圖；確定單元，用于確定所述待處理二分圖中的少數類節點以及多數類節點；分配單元，用于為各少數類節點分配對應各自的嵌入向量，為各多數類節點分配一個共享的嵌入向量，輸出所述待處理二分圖的向量分配結果。

根據本申請一優選實施例，所述確定單元在確定所述待處理二分圖中的少數類節點以及多數類節點時，具體執行：分別獲取所述待處理二分圖中兩個節點集合所包含節點的數量；將節點數量較少的節點集合中的各節點作為少數類節點，將另一節點集合中的各節點作為多數類節點。

根據本申請一優選實施例，所述分配單元還執行：確定與各多數類節點具有連接關系的鄰居節點；將對應同一個多數類節點的各鄰居節點對應的嵌入向量進行拼接，將拼接結果作為對應該多數類節點的嵌入向量。

根據本申請一優選實施例，所述分配單元在將對應同一個多數類節點的各鄰居節點對應的嵌入向量進行拼接時，具體執行：獲取各鄰居節點對應的嵌入向量；將各鄰居節點對應的嵌入向量與所述共享的嵌入向量進行拼接。

上述申請中的一個實施例具有如下優點或有益效果：本申請能夠提升嵌入向量的分配效率，降低為節點所分配的嵌入向量占用的內存資源。因為采用了只對二分圖中的部分節點分配各自所對應的嵌入向量的方式，所以克服了現有技術中對全部節點分配各自所對應的嵌入向量所導致的分配效率低、內存資源占用大的技術問題，達到提升嵌入向量的分配效率、降低嵌入向量所占用的內存資源的技術效果。