kooplearn.datasets

compute_prinz_potential_eig(gamma, sigma, dt)

Computes the eigenvalues and eigenfunctions of the Generator of the Overdamped Langevin dynamics for the Prinz potential.

compute_logistic_map_eig([M, eval_right_on, ...])

Compute eigenvalues and eigenfunctions of the Transfer operator.

compute_logistic_map_invariant_pdf([M, ...])

Compute the invariant probability density function.

fetch_ordered_mnist(*[, num_digits, ...])

Fetch the MNIST dataset and return an ordered subset interleaving samples from each digit class.

make_duffing(X0[, n_steps, dt, alpha, beta, ...])

Generate a trajectory from the Duffing oscillator.

make_linear_system(X0, A[, n_steps, noise, ...])

Generate a trajectory from a discrete-time linear dynamical system.

make_logistic_map(X0[, n_steps, r, M, dt, ...])

Generate a trajectory from the logistic map with optional trigonometric noise Ostruszka et al. [1].

make_lorenz63(X0[, n_steps, dt, sigma, mu, beta])

Generate a trajectory from the Lorenz-63 system.

make_prinz_potential(X0[, n_steps, dt, ...])

Generate a 1D Langevin trajectory for the "Prinz potential" Prinz et al. [1].

make_regime_switching_var(X0, phi1, phi2, ...)

Generate a trajectory from a regime-switching vector autoregressive (VAR) process.