添加边

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