Can agnn.py runnable with 'mag' or 'dblp' dataset

[pdhung3012]

The current agnn.py, in my understanding, is the implementation of a homogeneous GNN model for planetoid dataset. When I check this dataset, it only has nodes with no information about type and also the edges between nodes.

My question is, can the current agnn.py be available for support other datasets that have different types of nodes such as 'mag' and 'dblp'? To do that, do we need to remove information about types of nodes in these datasets?

Thank you

[rusty1s]

You should be able to utilize to_hetero to convert an AGNN network into its heterogeneous counterpart, see here. Let me know if this works for you :)