入门指南
自带
扩展 PyKEEN
参考
InductiveNodePieceGNN
pykeen.nn
附录
基础类:InductiveNodePiece
InductiveNodePiece
在GNN编码器之上的归纳式NodePiece。
总体而言,这是一个三步过程:
通过NodePiece对节点进行特征化
使用NodePiece特征在活动图上传递消息
给定一批三元组的评分函数
到目前为止,消息传递预计将在整个图上进行
初始化模型。
gnn_encoder (Iterable[Module] | None) – 一个消息传递层的可迭代对象。默认为具有Hadamard组合的2层CompGCN。
kwargs – 传递给 InductiveNodePiece.__init__ 的额外基于关键字的参数。
方法总结
reset_parameters_()
reset_parameters_
除了其他参数外,显式重置GNN编码器。
方法文档