TianxiangZhao / GraphSmote

Pytorch implementation of paper 'GraphSMOTE: Imbalanced Node Classification on Graphs with Graph Neural Networks' to appear on WSDM2021
150 stars 29 forks source link

Heterogeneous Graph Models #12

Open andrewbowen19 opened 5 months ago

andrewbowen19 commented 5 months ago

Hi there,

Thanks for putting GraphSMOTE together, really cool code! Was wondering if there was support for heterogeneous graph neural networks as well. I'm currently training a heterogeneous GNN (using SageCONV) on a dataset for a multi class classification problem. I didn't see anything in the paper/repo about support for heterogeneous GNNs, and wanted to check here. Thanks again!