microsoft / tf2-gnn

TensorFlow 2 library implementing Graph Neural Networks
MIT License
369 stars 73 forks source link

Can RGAT be implemented for a graph level multiclass-classification? #57

Open NipunaChhabra opened 2 years ago

NipunaChhabra commented 2 years ago

I have written a graph mutli-class classification task, following the lead from the GraphBinaryClassificationTask, and a dataset file following the lead from json_graph_property_dataset, but how do I use an rgat layer now? The PPI is an implementation of node level task, so I wanted to make sure what I'm trying to do is plausible here.