GCNNorm
- class dgl.transforms.GCNNorm(eweight_name='w')[source]
Bases:
BaseTransform
对输入图应用对称邻接归一化并保存结果边的权重,如使用图卷积网络进行半监督分类中所述。
对于异构图,这仅适用于对称的规范边类型,其源节点类型和目标节点类型是相同的。
- Parameters:
eweight_name (str, optional) –
edata
用于检索和存储边权重的名称。边权重是可选的。
示例
>>> import dgl >>> import torch >>> from dgl import GCNNorm >>> transform = GCNNorm() >>> g = dgl.graph(([0, 1, 2], [0, 0, 1]))
案例1:转换一个无权图
>>> g = transform(g) >>> print(g.edata['w']) tensor([0.5000, 0.7071, 0.0000])
案例2:转换一个加权图
>>> g.edata['w'] = torch.tensor([0.1, 0.2, 0.3]) >>> g = transform(g) >>> print(g.edata['w']) tensor([0.3333, 0.6667, 0.0000])