search

cszhangzhen / HGP-SL

Hierarchical Graph Pooling with Structure Learning
324 stars 54 forks source link

Use of this work for regression and classification #11

Open Bennyyy27 opened 3 years ago

Bennyyy27 commented 3 years ago

Hi! Great Work you have been doing. I am not a NN expert, but I think that the idea is major for many problems. I would like to use that GNN for testing something. However, due to my imbalanced dataset, the classification leads to a single class output. I guess I could change the type of this NN from classification to regression right? I already changed last linear layer to output 1 without activation and changed the input. However, i still get an error. Do you know if its possible to quickly change it to a regression problem?

Thank you

cszhangzhen commented 3 years ago

Hi,

It can be adapted to a regression problem. I'm wondering what your input is and what your output is.

By the way, what's your error information?