请问General Recommendation的边是否能设置权重？

[ralgond]

谢谢！

[hyp1231]

您好！抱歉回复较晚。目前已实现的几个通用图卷积模块（LightGCN, BipartiteGCNConv, BiGNNConv）都提供了输入边的权重 edge_weight 的接口，如

RecBole-GNN/recbole_gnn/model/layers.py

Lines 8 to 14 in 77d76b9

	class LightGCNConv(MessagePassing):
	def __init__(self, dim):
	super(LightGCNConv, self).__init__(aggr='add')
	self.dim = dim
	def forward(self, x, edge_index, edge_weight):
	return self.propagate(edge_index, x=x, edge_weight=edge_weight)

您可以新建模型时调用这些类，并传入您希望设置的边权重。

[ralgond]

“您可以新建模型时调用这些类，并传入您希望设置的边权重。”

我想请教一下新建模型的步骤，请问有没有参考文档和代码？

[hyp1231]

您好，新建模型可以参考这篇 API Doc，同时您也可以参考 https://github.com/RUCAIBox/RecBole-GNN/tree/main/recbole_gnn/model 下面对应类别模型的实现。