search

SakanaAI / AI-Scientist

The AI Scientist: Towards Fully Automated Open-Ended Scientific Discovery 🧑‍🔬
Apache License 2.0
8.26k stars 1.16k forks source link

Add SEIR model template #137

Closed speed1313 closed 1 month ago

speed1313 commented 1 month ago

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 https://github.com/SakanaAI/AI-Scientist/blob/8383c8a7d835a8b93a08dec50a93cd745242c1e1/ai_scientist/perform_experiments.py#L64 in the run_experiment() function needs to be changed to 

results = {k: v for k, v in results.items() if k != "solution"}
conglu1997 commented 1 month ago

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 commented 1 month ago

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