Expand description
Training utilities for Echo State Networks (ESN).
reservoir-train provides training-time components for Echo State Networks:
- ridge / lasso solvers for linear readouts
- an
ESNBuilderfor quickly assembling common ESN configurations - (optional,
std) static code generation for embedded inference models
This crate pairs with [reservoir-infer] (inference-time reservoirs/readouts).
§Features
std(default): Enablesstdand code generation utilities.libm: Enableslibm-backed math forno_stdtargets (also affects dependencies).
§High-level workflow
- Build an ESN (dense or sparse) with
ESNBuilder - Fit the readout using
ESNFitRidge::fitorESNFitLasso::fit_lasso - Run inference using the same model (
predict), or export weights for deployment.
§Examples
See the examples/ directory:
esn_henon.rsesn_mackey_glass.rsesn_narma.rs
Re-exports§
pub use codegen::StaticModelGenerator;pub use esn::ESNBuilder;pub use esn::ESNFitLasso;pub use esn::ESNFitRidge;pub use trainer::RidgeTrainer;pub use reservoir_infer;
Modules§
Structs§
- Dense
Reservoir - Dense ESN reservoir using dense matrices.
- Echo
State Network - A lightweight wrapper that wires a
Reservoirand aReadoutfor inference. - Lasso
Readout - Linear readout backed by a dense matrix (
DMatrix). - Ridge
Readout - Linear readout backed by a dense matrix (
DMatrix).
Type Aliases§
- RngType
- RNG type used internally by
ESNBuilder.