From c4c0a6e1cbfa69cb35daab894c2162d5c1ee5bfd Mon Sep 17 00:00:00 2001
From: Konoha <2363939673kai@gmail.com>
Date: Sat, 29 Nov 2025 01:12:50 +0800
Subject: [PATCH] handle isolated points

---
 gcn.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/gcn.py b/gcn.py
index 9513473..f29dd58 100644
--- a/gcn.py
+++ b/gcn.py
@@ -42,6 +42,8 @@ def aggregate(self,x, Adj):
             adj = adj-torch.eye(n).to(adj.device)
         if self.policy=='mean':
             num_neigh = adj.sum(1, keepdim=True)
+            # 修复：避免除以0（处理孤立节点）
+            num_neigh = torch.clamp(num_neigh, min=1.0)
             mask = adj.div(num_neigh)
             to_feats = mask.mm(x)
         elif self.policy=='max':