how to get the latent graph structure

qitianwu / NodeFormer

The official implementation of NeurIPS22 spotlight paper "NodeFormer: A Scalable Graph Structure Learning Transformer for Node Classification"

291 stars 28 forks source link

Closed bwrooney82 closed 2 months ago

bwrooney82 commented 3 months ago

i.e the adjacent matrix with weights of the latent graph

qitianwu commented 2 months ago

One can set return_weight as True here to obtain the latent structures produced by all-pair global attention in the model