Skip to main content

Framework for recursive Bayesian estimation in Python.

Project description

PyRecEst

Tests PyPI version Python versions Documentation

Recursive Bayesian Estimation for Python.

PyRecEst is a Python library for recursive Bayesian estimation on Euclidean spaces and manifolds. It uses a NumPy backend by default and can also run with PyTorch or JAX backends.

Features

PyRecEst provides tools for:

  • distributions and densities on Euclidean spaces and manifolds;
  • recursive Bayesian estimators, filters, and trackers;
  • multi-target tracking (MTT) and extended object tracking (EOT);
  • evaluation of filters and trackers; and
  • sampling distributions and generating grids.

Which API Should I Start With?

Task Start with
Linear Euclidean Gaussian filtering KalmanFilter and examples/basic/kalman_filter.py
Reusable transition and measurement models examples/basic/kalman_filter_with_models.py
Nonlinear filtering examples/basic/ukf_with_models.py or particle-filter examples
Multi-target tracking with clutter or missed detections examples/basic/multi_target_tracking.py
Circular, spherical, or manifold-valued states circular, hypertoroidal, and hyperspherical distribution modules
Backend-portable code pyrecest.backend imports plus focused tests under each backend

For more detail, see docs/choosing-an-api.md and docs/backend-compatibility.md.

Command Line And Reproducible Scenarios

After installation, use the lightweight CLI to inspect an environment or run a reproducible scenario:

pyrecest info
pyrecest backends --format markdown
pyrecest run-scenario scenarios/linear_gaussian_cv_1d/config.toml

The same scenario runner is available from a source checkout via scripts/run_scenario.py. Scenario definitions live in scenarios/ and include TOML configuration plus JSON golden outputs.

The pyrecest.diagnostics module defines common diagnostics containers for innovation statistics, particle-filter health, and association decisions.

Installation

PyRecEst requires Python 3.11 or newer and earlier than Python 3.15.

Install the package from PyPI:

python -m pip install pyrecest

Optional backend and domain-specific dependencies can be installed with extras:

python -m pip install "pyrecest[pytorch_support]"
python -m pip install "pyrecest[jax_support]"
python -m pip install "pyrecest[healpy_support]"
python -m pip install "pyrecest[all_support]"

Pip normalizes extra names, so the equivalent hyphenated spellings such as pyrecest[pytorch-support], pyrecest[jax-support], pyrecest[healpy-support], and pyrecest[all-support] may also appear in package-index metadata.

For development from a source checkout, use Poetry or the provided conda environment:

poetry install --with dev --all-extras
# or
conda env create -f environment.yml

Quickstart

The following example runs a one-dimensional constant-velocity Kalman filter. It uses the backend abstraction exposed by pyrecest.backend, so the same code can run on supported numerical backends.

from pyrecest.backend import array, diag
from pyrecest.filters import KalmanFilter


dt = 1.0
system_matrix = array([[1.0, dt], [0.0, 1.0]])
measurement_matrix = array([[1.0, 0.0]])
system_noise_cov = diag(array([0.05, 0.01]))
measurement_noise_cov = array([[0.25]])
measurements = [0.9, 2.0, 3.1, 3.9, 5.2]

kalman_filter = KalmanFilter((array([0.0, 1.0]), diag(array([1.0, 1.0]))))

for measurement in measurements:
    kalman_filter.predict_linear(system_matrix, system_noise_cov)
    kalman_filter.update_linear(
        array([measurement]), measurement_matrix, measurement_noise_cov
    )
    print(kalman_filter.get_point_estimate())

Run the complete script with:

python examples/basic/kalman_filter.py

Documentation

The docs/ directory contains the first project documentation pages:

  • Getting started covers installation, development setup, backend selection, and running examples.
  • API overview maps the main packages and points to the most common public entry points.
  • Backend compatibility explains the NumPy, PyTorch, and JAX support model and known limitations.
  • Choosing an API maps task families to recommended filters, distributions, trackers, and examples.
  • Distribution taxonomy summarizes the main representation families by domain and use case.
  • Error handling documents the shared exception types and user-facing diagnostics policy.
  • API reference contains generated package reference pages built with MkDocs and mkdocstrings.
  • Task tutorials show common distribution, filtering, tracking, and evaluation workflows.
  • Shapes and conventions documents common vector, matrix, measurement-set, batch, and manifold-coordinate shapes.
  • Examples lists the executable examples and what each one demonstrates.

Build the documentation site locally with:

poetry install --with docs --without dev
poetry run mkdocs build --strict

Backends

PyRecEst imports pyrecest.backend dynamically. The default backend is NumPy. Set PYRECEST_BACKEND before Python imports pyrecest to select another backend:

PYRECEST_BACKEND=pytorch python examples/basic/kalman_filter.py
PYRECEST_BACKEND=jax python examples/basic/kalman_filter.py

Install the matching optional extra before using a non-default backend.

Examples and tests

  • examples/basic/kalman_filter.py contains a small executable Kalman filter example.
  • examples/basic/scgp_measurement_reliability.py demonstrates reliability-weighted measurements for the full SCGP extended-object tracker.
  • tests/ contains additional usage examples for distributions, filters, smoothers, evaluation, sampling, metrics, and tracking utilities.

To run the test suite from a development environment:

python -m pytest

Citation

If you use PyRecEst in your research, please cite:

BibTeX BibLaTeX
@misc{pfaff_pyrecest_2023,
  author       = {Florian Pfaff},
  title        = {PyRecEst: Recursive Bayesian Estimation for Python},
  year         = {2023},
  howpublished = {\url{https://github.com/FlorianPfaff/PyRecEst}},
  note         = {MIT License}
}
@software{pfaff_pyrecest_2023_software,
  author    = {Florian Pfaff},
  title     = {PyRecEst: Recursive Bayesian Estimation for Python},
  year      = {2023},
  url       = {https://github.com/FlorianPfaff/PyRecEst},
  license   = {MIT},
  keywords  = {Bayesian filtering; manifolds; tracking; Python; NumPy; PyTorch; JAX}
}

Credits

PyRecEst borrows its structure from libDirectional and follows its code closely for many classes. libDirectional, a project to which Florian Pfaff contributed extensively, is available on GitHub. The backend implementations are based on those of geomstats.

License

PyRecEst is licensed under the MIT License.

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

pyrecest-2.2.4.tar.gz (635.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

pyrecest-2.2.4-py3-none-any.whl (858.1 kB view details)

Uploaded Python 3

File details

Details for the file pyrecest-2.2.4.tar.gz.

File metadata

  • Download URL: pyrecest-2.2.4.tar.gz
  • Upload date:
  • Size: 635.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for pyrecest-2.2.4.tar.gz
Algorithm Hash digest
SHA256 9534fee6701a3ded8383227597275cd1038f77a7fae3c9e1d925dc92089a1588
MD5 8e047093c7e528ceab5e3136eebe4999
BLAKE2b-256 d8eb579505f2f2429dd9edfda267a49d7dab50cc9756157ebfe71e44ee3a75f7

See more details on using hashes here.

Provenance

The following attestation bundles were made for pyrecest-2.2.4.tar.gz:

Publisher: publish-to-pypi.yml on FlorianPfaff/PyRecEst

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file pyrecest-2.2.4-py3-none-any.whl.

File metadata

  • Download URL: pyrecest-2.2.4-py3-none-any.whl
  • Upload date:
  • Size: 858.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for pyrecest-2.2.4-py3-none-any.whl
Algorithm Hash digest
SHA256 825aa107c03aeb538e32a303db2673e6225b1039af43b84a78a65ef4926a6f75
MD5 c477e928cae97953fd14d9f5dbd51f22
BLAKE2b-256 98abb1a3fa2a38da4354671927f1ed35de4263e463cfb148b06c19c06e3a5eb3

See more details on using hashes here.

Provenance

The following attestation bundles were made for pyrecest-2.2.4-py3-none-any.whl:

Publisher: publish-to-pypi.yml on FlorianPfaff/PyRecEst

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page