AddEdge

class dgl.transforms.AddEdge(ratio=0.2)[source]

基类:BaseTransform

随机添加边,如 Graph Contrastive Learning with Augmentations 论文中所述。

参数:

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