search

lrjconan / GRAN

Efficient Graph Generation with Graph Recurrent Attention Networks, Deep Generative Model of Graphs, Graph Neural Networks, NeurIPS 2019
MIT License
462 stars 97 forks source link

Attributes #9

Closed Kremlesis closed 4 years ago

Kremlesis commented 4 years ago

How to get the attributes of nodes and edges in the generated graphs. The current output only contains the adjacency matrix of the graphs. Is there any way we can get the nodes and edges attribute matrix also? Thanks

lrjconan commented 4 years ago

How to get the attributes of nodes and edges in the generated graphs. The current output only contains the adjacency matrix of the graphs. Is there any way we can get the nodes and edges attribute matrix also? Thanks

For the attributes, you can add prediction heads that take the node representations output by the GNN as the input.