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

ARMAConv Hyperparametertuning #1078

Open
3 of 4 tasks
AgathaSchmidt opened this issue Jul 29, 2024 · 0 comments
Open
3 of 4 tasks

ARMAConv Hyperparametertuning #1078

AgathaSchmidt opened this issue Jul 29, 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 29, 2024

Enhancement description

As the GNN grid search from #1070 showed, the best performing GNN is an ARMAConv model, based on the ARMAConv layer provided by the python library Spektral https://graphneural.network/layers/convolution/#armaconv

As we can see in the documentation, there are several layer specific parameters that can be altered. In order to find the best architecture for our task of predicting the diseasy dynamic for all 400 counties we conduct experiments, chinging the model parameters of the ARMAConv model.

  • hyperparameter tuning for ARMAConv (model specific parameters)
  • tuning hyperparameters like activation function and optimizer

requirements

GPU Quadro RTX 4000
tensorflow 2.9.1
numpy 1.22 4
scikit-learn 1.5.1
spektral 1.3.1
keras 2.14.0

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 29, 2024
@AgathaSchmidt AgathaSchmidt self-assigned this Jul 29, 2024
@AgathaSchmidt AgathaSchmidt moved this from Product Backlog to 🏗 Development in MEmilio: Equation-Based-Models Development Aug 7, 2024
@AgathaSchmidt AgathaSchmidt moved this from 📋 Product backlog to 🏗 In progress in MEmilio: Surrogate-Modeling with Machine Learning Aug 7, 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