在全新图形上使用StellarGraph图形注意力网络

时间:2020-08-18 18:09:11

标签: python tensorflow graph neural-network

我已经使用Stellargraph和tensorflow在图(119468)节点上训练了图注意力网络。如何使用Stellargraph API使用全新的图形进行节点级别的预测(分类)?根据{{​​3}}

使用Stellargraphs FullBatchNodeGenerator对图形进行了训练

我正在尝试使用新实例化的FullBatchNodeGenerator并将新图作为输入(1746个节点)来预测新图上的节点,但这会引发错误:

InvalidArgumentError:  Cannot multiply A and B because inner dimension does not match: 119468 vs. 1746.  Did you forget a transpose?  Dimensions of A: [119468, 119468).  Dimensions of B: [1746,8]

使用Stellargraph的生成器和API是否有可能?

谢谢

0 个答案:

没有答案
相关问题