Open ralgond opened 1 year ago
您好!抱歉回复较晚。目前已实现的几个通用图卷积模块(LightGCN
, BipartiteGCNConv
, BiGNNConv
)都提供了输入边的权重 edge_weight
的接口,如
您可以新建模型时调用这些类,并传入您希望设置的边权重。
“您可以新建模型时调用这些类,并传入您希望设置的边权重。”
我想请教一下新建模型的步骤,请问有没有参考文档和代码?
您好,新建模型可以参考这篇 API Doc,同时您也可以参考 https://github.com/RUCAIBox/RecBole-GNN/tree/main/recbole_gnn/model 下面对应类别模型的实现。
谢谢!