search

epirecipes / sir-julia

Various implementations of the classical SIR model in Julia
MIT License
196 stars 39 forks source link

Add NNODE example #75

Open sdwfrost opened 1 year ago

sdwfrost commented 1 year ago

Currently stuck on adding an NNODE - an important prerequisite for using NeuralPDE.jl to fit PDE models. See https://discourse.julialang.org/t/constraints-in-neuralpde-nnode/92723/2 for a toy example that fails.

ChrisRackauckas commented 1 year ago

Yeah the simple MLP neural networks are usually not sufficient. I'm not sure if that's just a documentation issue or something we can handle more easily within the package itself, given that NNODE is a constrained form. Maybe it should default to an architecture we know well and only give options for layer sizes.