Skip to main content

DeePyMoD is a PyTorch-based implementation of the DeepMoD algorithm for model discovery of PDEs.

Project description



DeePyMoD is a modular framework for model discovery of PDEs and ODEs from noise data. The framework is comprised of four components, that can seperately be altered: i) A function approximator to construct a surrogate of the data, ii) a function to construct the library of features, iii) a sparse regression algorithm to select the active components from the feature library and iv) a constraint on the function approximator, based on the active components.


More information can be found in the following two papers: , arXiv:2011.04336, arXiv:1904.09406 and the full documentation is availeble on

What's the use case? Classical Model Discovery methods struggle with elevated noise levels and sparse datasets due the low accuracy of numerical differentiation. DeepMoD can handle high noise and sparse datasets, making it well suited for model discovery on actual experimental data.

What types of models can you discover? DeepMoD can discover non-linear, multi-dimensional and/or coupled ODEs and PDEs. See our paper and the examples folder for a demonstration of each.

How to install

Dependencies and CUDA

We support Python 3.6, 3.7 and 3.8. We rely on the following packages, they will be installed in the pip installation proces for you: numpy, torch, sklearn, pysindy, natsort, tensorboard, matplotlib

We also make use of the PyTorch library, which can be installed with CPU and/or GPU support. Please refer to the PyTorch "Get Started" guide to get the version that is optimal for your system.


To install DeePyMoD, pip can be used

pip install deepymod

in the main directory.


If you wish to alter the code you can clone the package using:

git clone

and then install it from the cloned DeePyMoD directory using

pip install -e ./


  • Many example notebooks We have implemented a varyity of examples ranging from 2D Advection Diffusion, Burgers' equation to non-linear, higher order ODE's If you miss any example, don't hesitate to give us a heads-up.

  • Extendable DeePyMoD is designed to be easily extendable and modifiable. You can simply plug in your own cost function, library or training regime.

  • Automatic library The library and coefficient vectors are automatically constructed from the maximum order of polynomial and differentiation. If that doesn't cut it for your use case, it's easy to plug in your own library function.

  • Extensive logging We provide a simple command line logger to see how training is going and an extensive custom Tensorboard logger.

  • Fast Depending on the size of the data-set DeepMoD, running a model search with DeepMoD takes of the order of minutes/ tens of minutes on a standard CPU. Running the code on GPU's drastically improves performence.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

DeePyMoD-2.1.0.tar.gz (4.6 MB view hashes)

Uploaded source

Built Distribution

DeePyMoD-2.1.0-py2.py3-none-any.whl (29.8 kB view hashes)

Uploaded py2 py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page