controlled-sde-learn is a Python library designed to estimate the coefficients of controlled stochastic differential equations (SDEs). The approach leverages kernel methods and Fokker-Planck equation matching to estimate the drift and diffusion coefficients from a data set of controlled sample paths with random controls.
The examples
folder contains several scripts demonstrating different applications of the controlled-sde-learn
library.
- example_ornstein_uhlenbeck_paths_plot.py. Illustrates the generation and plotting of sample paths from a controlled Ornstein-Uhlenbeck process.
- example_dubins_paths_plot.py. Illustrates the generation and plotting of sample paths from a controlled Dubins process.
- example_kde_plot.py. Demonstrates the use of the
ProbaDensityEstimator
for estimating and visualizing the probability density of sample paths from a controlled SDE under different controls. - example_sde_identification_1d.py. Provides a complete example for simulating a one-dimensional controlled SDE and estimating its coefficients using Fokker-Planck matching.
- example_sde_identification_2d.py. Presents a complete example for estimating the coefficients of a two-dimensional nonlinear controlled SDE.
Explore these examples to understand how to simulate controlled SDEs and estimate their coefficients.
To install:
- Clone the repository.
git clone https://github.com/lmotte/controlled-sde-learn.git
- Install the required dependencies (Python 3.x required).
pip install -r requirements.txt
This project is licensed under the MIT License - see the LICENSE file for details.