Use of this work for regression and classification

[Bennyyy27]

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]

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?