-
Notifications
You must be signed in to change notification settings - Fork 314
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
Could Cogdl implement graph regression task? #342
Comments
Hi @huang429, |
@cenyk1230感谢您的回复。 class MLPregression(nn.Module):
2.我想要将图分类预测(如GIN模型)转化为图回归预测,在GIN最后一层加了MLPRegression。 |
I used MLP as the regression layer,but got an error that
ValueError: Expected 2 or more dimensions (got 1)
Is that couldn't implement graph regression task,please?
The text was updated successfully, but these errors were encountered: