AddEdgeΒΆ
-
class
dgl.transforms.
AddEdge
(ratio=0.2)[source]ΒΆ Bases:
dgl.transforms.module.BaseTransform
Randomly add edges, as described in Graph Contrastive Learning with Augmentations.
- Parameters
ratio (float, optional) β Number of edges to add divided by the number of existing edges.
Example
>>> import dgl >>> from dgl import AddEdge
>>> transform = AddEdge() >>> g = dgl.rand_graph(5, 20) >>> new_g = transform(g) >>> print(new_g.num_edges()) 24