Getting started

Install

pip install echostatenetwork          # core (numpy/scipy/matplotlib)
pip install "echostatenetwork[opt]"   # + scikit-optimize for Bayesian hyperparameter search
pip install -e ".[dev]"               # development

Minimal workflow

import numpy as np
from echostatenetwork import EchoStateNetwork

y = my_time_series                       # (Nt, N_dim)

esn = EchoStateNetwork(y=y[:1].T,        # sets the state dimension
                       dt=0.01,          # time step of the data
                       N_units=200,      # reservoir size
                       t_train=40.0,     # training window
                       t_val=4.0)        # validation window
esn.train(y)                             # ridge regression + Bayesian hyperparameter search

Training accepts a single trajectory (Nt, N_dim), a batch of segments (L, Nt, N_dim), or a ragged list of segments of different lengths. After training, esn.step(u, r) advances the reservoir one step; washout (open loop on observed data) followed by closed-loop prediction is shown end to end in the tutorials.

Parametric ESN

Pass input_parameters (shape (N_param, L), one parameter vector per training segment) to condition the reservoir on a physical parameter — see the parametric ESN tutorial.