添加边
- class dgl.transforms.AddEdge(ratio=0.2)[source]
Bases:
BaseTransform
随机添加边,如Graph Contrastive Learning with Augmentations中所述。
- Parameters:
ratio (float, optional) – 要添加的边数除以现有边数。
示例
>>> import dgl >>> from dgl import AddEdge
>>> transform = AddEdge() >>> g = dgl.rand_graph(5, 20) >>> new_g = transform(g) >>> print(new_g.num_edges()) 24