ezmsg-learn

This repository contains a Python package with modules for machine learning (ML)-related processing in the ezmsg framework. As ezmsg is intended primarily for processing unbounded streaming signals, so are the modules in this repo.

If you are only interested in offline analysis without concern for reproducibility in online applications, then you should probably look elsewhere.

Processing units include dimensionality reduction, linear regression, and classification that can be initialized with known weights, or adapted on-the-fly with incoming (labeled) data.

Installation

The base install is NumPy-only. The machine-learning backends are optional extras, so a deployment that uses only the lightweight processors does not pay for a PyTorch or scikit-learn install:

pip install ezmsg-learn              # numpy-only processors
pip install "ezmsg-learn[sklearn]"   # + pandas, river, scikit-learn
pip install "ezmsg-learn[torch]"     # + torch
pip install "ezmsg-learn[all]"       # everything

Or install the latest development version:

pip install "git+https://github.com/ezmsg-org/ezmsg-learn@dev#egg=ezmsg-learn[all]"

Importing a module whose backend is not installed raises an ImportError naming the extra to install.

Dependencies

Base (pip install ezmsg-learn) — ezmsg, ezmsg-baseproc, ezmsg-sigproc, numpy, scipy, array-api-compat.

Extra	Adds	Covers
(none)	—	`process.ssr`, `process.flatten`, `process.seqseqsampler`, `process.refit_kalman`, `model.cca`, `model.refit_kalman`
`sklearn`	`pandas`, `river`, `scikit-learn`	`process.adaptive_linear_regressor`, `process.linear_regressor`, `process.sgd`, `process.slda`, `process.sklearn`, `dim_reduce.*`
`torch`	`torch`	`process.base`, `process.torch`, `process.rnn`, `process.transformer`, `process.mlp_old`, `model.mlp`, `model.rnn`, `model.transformer`
`all`	both of the above	everything, including all `collection.sample_adapt_regressor` backends

collection.sample_adapt_regressor imports its backend lazily, so it needs only the extra for the model_type in use — and none at all for model_type="kalman".

Development

We use uv for development.

Install uv if not already installed.
Fork this repository and clone your fork locally.
Open a terminal and cd to the cloned folder.
Run uv sync to create a .venv and install dependencies.
(Optional) Install pre-commit hooks: uv run pre-commit install
After making changes, run the test suite: uv run pytest tests

License

MIT License - see LICENSE for details.

Download files

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

Source Distribution

ezmsg_learn-1.13.0.tar.gz (106.9 kB view details)

Uploaded Aug 24, 2026 Source

Built Distribution

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

The dropdown lists show the available interpreters, ABIs, and platforms. Enable javascript to be able to filter the list of wheel files.

ezmsg_learn-1.13.0-py3-none-any.whl (74.5 kB view details)

Uploaded Aug 24, 2026 Python 3

File details

Details for the file ezmsg_learn-1.13.0.tar.gz.

File metadata

Download URL: ezmsg_learn-1.13.0.tar.gz
Upload date: Aug 24, 2026
Size: 106.9 kB
Tags: Source
Uploaded using Trusted Publishing? Yes
Uploaded via: uv/0.12.5 {"installer":{"name":"uv","version":"0.12.5","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

File hashes

Hashes for ezmsg_learn-1.13.0.tar.gz
Algorithm	Hash digest
SHA256	`42b9a9f85e824052ed1c9fa51afe9227a54ef30e7f30aefbfdb9749290156fef`
MD5	`d7d46be53757c0ed6030f986a3967404`
BLAKE2b-256	`86b0a69d821077f307f04d60c1fbeb46188962ece3fb146c4f0fe755b7e72f2d`

See more details on using hashes here.

File details

Details for the file ezmsg_learn-1.13.0-py3-none-any.whl.

File metadata

Download URL: ezmsg_learn-1.13.0-py3-none-any.whl
Upload date: Aug 24, 2026
Size: 74.5 kB
Tags: Python 3
Uploaded using Trusted Publishing? Yes
Uploaded via: uv/0.12.5 {"installer":{"name":"uv","version":"0.12.5","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

File hashes

Hashes for ezmsg_learn-1.13.0-py3-none-any.whl
Algorithm	Hash digest
SHA256	`b27eb88e85284f28e711cb7046dc25fac4d622687c8e338da020757efe3c414d`
MD5	`f80d101c534da35fd042fa573e417330`
BLAKE2b-256	`dd1535b3b0d3f0967b74669c55e248e12507b4dc41459ba485d81fc5fb2cb402`