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

Grid Search for GNNs #1070

Open
3 tasks done
AgathaSchmidt opened this issue Jul 18, 2024 · 0 comments
Open
3 tasks done

Grid Search for GNNs #1070

AgathaSchmidt opened this issue Jul 18, 2024 · 0 comments
Assignees
Labels
class::feature A feature to be implemented for some part of the software class::improvement Cleanup that doesn't affect functionality model::ode This issue concerns any kind of ODE-based model.

Comments

@AgathaSchmidt
Copy link
Member

AgathaSchmidt commented Jul 18, 2024

Enhancement description

In order to find the best GNN architecture suitable for our task, we conduct a grid serach for foru different model types.
Tha parameters that are varied are the model depth (number of layers) and the model width (number of units per layer).
This code is important to ensure replicability of the results of the master thesis.

The training data is generated by the MEmilio graph-ODE model and then split into input and label for the model as implemented in #1057.

  • grid serach for four model types (GCNConv, ARMAConv, APPNPConv, GATConv)

Checklist

  • Attached labels, especially loc:: or model:: labels.
  • Linked to project
@AgathaSchmidt AgathaSchmidt added class::feature A feature to be implemented for some part of the software model::ode This issue concerns any kind of ODE-based model. class::improvement Cleanup that doesn't affect functionality labels Jul 18, 2024
@AgathaSchmidt AgathaSchmidt self-assigned this Jul 18, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
class::feature A feature to be implemented for some part of the software class::improvement Cleanup that doesn't affect functionality model::ode This issue concerns any kind of ODE-based model.
Projects
Status: 🏗 Development
Development

When branches are created from issues, their pull requests are automatically linked.

1 participant