-
Notifications
You must be signed in to change notification settings - Fork 53
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Use of this work for regression and classification #11
Comments
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? |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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
The text was updated successfully, but these errors were encountered: