归纳节点片段图神经网络

class InductiveNodePieceGNN(*, gnn_encoder: Iterable[Module] | None = None, **kwargs)[来源]

基础类:InductiveNodePiece

在GNN编码器之上的归纳式NodePiece。

总体而言,这是一个三步过程:

  1. 通过NodePiece对节点进行特征化

  2. 使用NodePiece特征在活动图上传递消息

  3. 给定一批三元组的评分函数

到目前为止,消息传递预计将在整个图上进行

初始化模型。

Parameters:
  • gnn_encoder (Iterable[Module] | None) – 一个消息传递层的可迭代对象。默认为具有Hadamard组合的2层CompGCN。

  • kwargs – 传递给 InductiveNodePiece.__init__ 的额外基于关键字的参数。

方法总结

reset_parameters_()

除了其他参数外,显式重置GNN编码器。

方法文档

reset_parameters_()[来源]

除了其他参数外,显式重置GNN编码器。