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.

Sign up for GitHub

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

Implement UGraphEmb pooling #1763

Draft

huonw wants to merge 1 commit into develop from feature/1627-ugraphemb-pooling

Member

huonw commented Jun 30, 2020

This implements the pooling layer from UGraphEmb. This doesn't quite fully implement the algorithm:

the output of each GCN/GIN layer needs to be pooled separately
the loss function seems to become nan when training with this in the existing unsupervised graph classification algorithm, so this clearly isn't fully useable yet

See: #1627


Implement UGraphEmb pooling

1f13648

codeclimate bot commented Jun 30, 2020

Code Climate has analyzed commit 1f13648 and detected 0 issues on this pull request.

View more on Code Climate.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment