pyg-team
★ 23,960pytorch_geometric
PyTorch 的圖神經網路庫。為此,我們載入 Cora 資料集,並使用預先定義的 GCNConv 建立一個簡單的 2 層 GCN 模型:我們現在可以在訓練循環中最佳化模型,類似於標準 PyTorch 訓練過程。
HYSEN LABS 專案目錄
來自 pyg-team 組織的可信儲存庫、分類與深度解析。
PyTorch 的圖神經網路庫。為此,我們載入 Cora 資料集,並使用預先定義的 GCNConv 建立一個簡單的 2 層 GCN 模型:我們現在可以在訓練循環中最佳化模型,類似於標準 PyTorch 訓練過程。