torch_geometric.transforms.OneHotDegree

class OneHotDegree(max_degree: int, in_degree: bool = False, cat: bool = True)[source]

基础类：BaseTransform

将节点度数作为独热编码添加到节点特征中 (函数名称: one_hot_degree).

Parameters:

  • max_degree (int) – 最大度数。

  • in_degree (bool, 可选) – 如果设置为 True，将计算节点的入度而不是出度。 (默认: False)

  • cat (bool, 可选) – 将节点度数连接到节点特征，而不是替换它们。（默认值：True）