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

Add SEIR model template #137

Merged
merged 4 commits into from
Oct 24, 2024
Merged

Conversation

speed1313
Copy link
Contributor

Thank you for your great and interesting work!

I prepared a custom template of the SEIR model, a differential equation model for predicting infectious diseases.
For more details, please visit my blog post, where I wrote about an experiment using this template.

Note
To ensure this template code runs correctly, the line

results = {k: v["means"] for k, v in results.items()}

in the run_experiment() function needs to be changed to

results = {k: v for k, v in results.items() if k != "solution"}

@speed1313 speed1313 changed the title add SEIR model template Add SEIR model template Oct 9, 2024
@conglu1997 conglu1997 added the enhancement New feature or request label Oct 22, 2024
@conglu1997
Copy link
Collaborator

Would you be able to add a dummy means field for your new variable so we don't have to make a change that could be hard to maintain in the future? Thanks so much!

@speed1313
Copy link
Contributor Author

Thank you for reviewing!
I added a dummy means and a dummy dataset name. We can run this template without any modification now.

@conglu1997 conglu1997 merged commit 399c6be into SakanaAI:main Oct 24, 2024
@speed1313 speed1313 deleted the add-seir-template branch October 24, 2024 10:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants