随机游走PE

class dgl.transforms.RandomWalkPE(k, feat_name='PE', eweight_name=None)[source]

Bases: BaseTransform

随机游走位置编码，如具有可学习结构和位置表示的图神经网络中介绍

该模块仅适用于同构图。

Parameters:

  • k (int) – 随机游走的步数。论文发现，在两个实验中，最佳值为16和20。

  • feat_name (str, optional) – 用于存储计算出的位置编码在ndata中的名称。

  • eweight_name (str, optional) – 用于检索边权重的名称。默认值：None，不使用边权重。

示例

>>> import dgl
>>> from dgl import RandomWalkPE

>>> transform = RandomWalkPE(k=2)
>>> g = dgl.graph(([0, 1, 1], [1, 1, 0]))
>>> g = transform(g)
>>> print(g.ndata['PE'])
tensor([[0.0000, 0.5000],
        [0.5000, 0.7500]])