Skip to content
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

Allow using arbitrary edge features in GNNs #9

Merged
merged 4 commits into from
Feb 22, 2021

Conversation

mallamanis
Copy link
Contributor

No description provided.

@mallamanis mallamanis self-assigned this Feb 15, 2021
Copy link
Contributor

@mmjb mmjb left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

In principle this seems fine (see my minor comments). One thing I was wondering about was if it would make sense to support the case of having edge features only for some edges explicitly, but it's probably easier to ignore this for now (or just use constant values for those edges)

ptgnn/baseneuralmodel/abstractneuralmodel.py Show resolved Hide resolved
ptgnn/implementations/typilus/train.py Show resolved Hide resolved
ptgnn/neuralmodels/gnn/graphneuralnetwork.py Show resolved Hide resolved
ptgnn/neuralmodels/gnn/graphneuralnetwork.py Outdated Show resolved Hide resolved
@mallamanis mallamanis requested a review from mmjb February 22, 2021 09:52
@mallamanis mallamanis merged commit 7a23ce4 into master Feb 22, 2021
@mallamanis mallamanis deleted the dev/miallama/edge-features branch February 22, 2021 10:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants