圖神經網絡入門示例:使用PyTorch Geometric 進行節點分類

基於圖的神經網絡是強大的模型,可以學習網絡中的複雜模式。在本文中,我們將介紹如何爲同構圖數據構造PyTorch Data對象,然後訓練不同類型的神經網絡來預測節點所屬的類。這種類型的預測問題通常被稱爲節點分類。

我們將使用來自Benedek Rozemberczki, Carl Allen和Rik Sarkar於2019年發佈的“Multi-scale Attributed Node Embedding”論文中的Facebook Large Page-Page Network¹數據集。

該數據集包含22,470個Facebook頁面,按主題分爲四類。由不同大小的特徵向量表示。數據集還包含Facebook pages 上跟隨其他page的信息。網絡中有171,992個鏈接或邊。

 

https://avoid.overfit.cn/post/885ad3f5eb424045b02408699c45e340

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章