我想在Pytorch Geometric中处理加权无向图。节点特征是50维的。我发现这可以通过torch_geometric.data.data
类的x
属性来处理。边的权重是标量值。我们发现edge_attr
和edge_weight
是处理边的属性。
我想我应该使用edge_weight
,这是正确的吗?
另外,edge_attr
和edge_weight
的区别是什么?
谢谢。
edge_weight
和edge_attr
的区别在于edge_weight
始终是一维的(每条边一个值),而edge_attribute
可以是多维的。您可以检查备忘单是否支持这些模型。
edge_weight
和edge_attr
的区别在于,edge_weight
是连接两个节点的边的非二进制表示,没有edge_weight
,连接两个节点的边要么存在要么不存在(0或1),但有了权值,连接节点的边可以是任意值。
而edge_attr
表示连接任意两个节点的边的特征。这些特征可以是多维的。