GNN邻接矩阵归一化

诸神缄默不语-个人CSDN博文目录

特征归一化见这篇博文:特征工程/数据预处理超全面总结(持续更新ing…)

文章目录

1. 对称归一化 D − 1 2 A D − 1 2 D^{-\frac{1}{2}}AD^{-\frac{1}{2}} D21AD21

代码参考自R-former:

def normalize_adj(adj):"""Symmetrically normalize adjacency matrix."""rowsum = np.array(adj.sum(1))with np.errstate(divide='ignore'):d_inv_sqrt = np.power(rowsum, -0.5).flatten()d_inv_sqrt[np.isinf(d_inv_sqrt)] = 0.d_mat_inv_sqrt = np.diag(d_inv_sqrt)return adj.dot(d_mat_inv_sqrt).transpose().dot(d_mat_inv_sqrt)

2. D − 1 A D^{-1}A D1A

代码参考自GL-GIN:

def normalize_adj(mx):"""Row-normalize matrix  D^{-1}Atorch.diag_embed: https://github.com/pytorch/pytorch/pull/12447"""mx = mx.float()rowsum = mx.sum(2)r_inv = torch.pow(rowsum, -1)r_inv[torch.isinf(r_inv)] = 0.r_mat_inv = torch.diag_embed(r_inv, 0)mx = r_mat_inv.matmul(mx)return mx


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部