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':