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

GraphSMOTE on dgl graph #10

Open Bai1026 opened 11 months ago

Bai1026 commented 11 months ago

Hello! Can I use this technique on the dgl graph with my own dataset? Would there be a lot of modification I need to de? I am now doing a edge classification task and encountered a data imbalance problem. The graph with only one edge can not be correctly detected, so I am also wondering if GraphSMOTE can help me deal with this problem.