Index

A

auto_batch() (in module ssm.utils)

AutoregressiveEmissions (class in ssm.arhmm.emissions) AutoregressiveHMM (class in ssm.arhmm.base)

B

BernoulliEmissions (class in ssm.hmm.emissions)

BernoulliHMM (class in ssm.hmm.models)

C

check_pytree_dtype_match() (in module ssm.utils) check_pytree_match() (in module ssm.utils) check_pytree_shape_match() (in module ssm.utils)

check_pytree_structure_match() (in module ssm.utils) check_pytree_weak_type_match() (in module ssm.utils) compute_state_overlap() (in module ssm.utils)

D

debug_rejit() (in module ssm.utils) distribution() (ssm.arhmm.emissions.AutoregressiveEmissions method) (ssm.factorial_hmm.emissions.NormalFactorialEmissions method) (ssm.factorial_hmm.initial.FactorialInitialCondition method) (ssm.factorial_hmm.transitions.FactorialTransitions method) (ssm.hmm.emissions.Emissions method) (ssm.hmm.emissions.ExponentialFamilyEmissions method) (ssm.hmm.initial.InitialCondition method) (ssm.hmm.initial.StandardInitialCondition method) (ssm.hmm.transitions.SimpleStickyTransitions method) (ssm.hmm.transitions.StationaryTransitions method) (ssm.hmm.transitions.Transitions method) (ssm.lds.dynamics.Dynamics method) (ssm.lds.dynamics.StationaryDynamics method) (ssm.lds.emissions.Emissions method) (ssm.lds.emissions.GaussianEmissions method) (ssm.lds.emissions.PoissonEmissions method) (ssm.lds.initial.InitialCondition method) (ssm.lds.initial.StandardInitialCondition method) (ssm.twarhmm.emissions.TimeWarpedAutoregressiveEmissions method)

Dynamics (class in ssm.lds.dynamics) dynamics_distribution() (ssm.base.SSM method) (ssm.hmm.base.HMM method) (ssm.lds.base.LDS method)

E

e_step() (ssm.lds.models.GaussianLDS method) elbo() (ssm.base.SSM method) em() (in module ssm.inference.em) Emissions (class in ssm.hmm.emissions) (class in ssm.lds.emissions) emissions_distribution() (ssm.arhmm.base.AutoregressiveHMM method) (ssm.base.SSM method) (ssm.hmm.base.HMM method) (ssm.lds.base.LDS method)

emissions_shape (ssm.base.SSM property) (ssm.hmm.base.HMM property) (ssm.lds.base.LDS property) ensure_has_batch_dim() (in module ssm.utils) ExponentialFamilyEmissions (class in ssm.hmm.emissions)

F

FactorialEmissions (class in ssm.factorial_hmm.emissions) FactorialHMM (class in ssm.factorial_hmm.base) FactorialInitialCondition (class in ssm.factorial_hmm.initial) FactorialTransitions (class in ssm.factorial_hmm.transitions)

find_permutation() (in module ssm.utils) fit() (ssm.hmm.base.HMM method) (ssm.lds.base.LDS method) (ssm.lds.models.GaussianLDS method)

G

GaussianARHMM (class in ssm.arhmm.models) GaussianEmissions (class in ssm.hmm.emissions) (class in ssm.lds.emissions)

GaussianHMM (class in ssm.hmm.models) GaussianLDS (class in ssm.lds.models) GaussianTWARHMM (class in ssm.twarhmm.models) gradient_cmap() (in module ssm.plots)

H

HMM (class in ssm.hmm.base)

I

initial_distribution() (ssm.base.SSM method) (ssm.hmm.base.HMM method) (ssm.lds.base.LDS method) InitialCondition (class in ssm.hmm.initial) (class in ssm.lds.initial)

initialize() (ssm.factorial_hmm.base.FactorialHMM method) (ssm.hmm.base.HMM method) (ssm.lds.base.LDS method) (ssm.twarhmm.models.GaussianTWARHMM method)

L

laplace_approximation() (in module ssm.inference.laplace_em) laplace_em() (in module ssm.inference.laplace_em) LDS (class in ssm.lds.base) log_initial_probs() (ssm.factorial_hmm.initial.FactorialInitialCondition method) (ssm.hmm.initial.InitialCondition method) (ssm.hmm.initial.StandardInitialCondition method) log_likelihoods() (ssm.arhmm.emissions.AutoregressiveEmissions method) (ssm.factorial_hmm.emissions.NormalFactorialEmissions method) (ssm.hmm.emissions.Emissions method) (ssm.twarhmm.emissions.TimeWarpedAutoregressiveEmissions method)

log_probability() (ssm.arhmm.base.AutoregressiveHMM method) (ssm.base.SSM method) log_transition_matrices() (ssm.factorial_hmm.transitions.FactorialTransitions method) (ssm.hmm.transitions.SimpleStickyTransitions method) (ssm.hmm.transitions.StationaryTransitions method) (ssm.hmm.transitions.Transitions method) logspace_tensordot() (in module ssm.utils)

M

m_step() (ssm.arhmm.emissions.AutoregressiveEmissions method) (ssm.factorial_hmm.initial.FactorialInitialCondition method) (ssm.factorial_hmm.transitions.FactorialTransitions method) (ssm.hmm.emissions.Emissions method) (ssm.hmm.emissions.ExponentialFamilyEmissions method) (ssm.hmm.initial.InitialCondition method) (ssm.hmm.initial.StandardInitialCondition method) (ssm.hmm.transitions.SimpleStickyTransitions method) (ssm.hmm.transitions.StationaryTransitions method) (ssm.hmm.transitions.Transitions method) (ssm.lds.base.LDS method) (ssm.lds.dynamics.Dynamics method) (ssm.lds.dynamics.StationaryDynamics method) (ssm.lds.emissions.Emissions method) (ssm.lds.emissions.GaussianEmissions method) (ssm.lds.emissions.PoissonEmissions method) (ssm.lds.initial.StandardInitialCondition method) (ssm.twarhmm.emissions.TimeWarpedAutoregressiveEmissions method) marginal_likelihood() (ssm.lds.models.GaussianLDS method) module ssm.arhmm.base ssm.arhmm.emissions ssm.arhmm.models ssm.base ssm.factorial_hmm.base ssm.factorial_hmm.emissions ssm.factorial_hmm.initial ssm.factorial_hmm.models ssm.factorial_hmm.transitions ssm.hmm.base ssm.hmm.emissions ssm.hmm.initial ssm.hmm.models ssm.hmm.transitions ssm.inference.em ssm.inference.laplace_em ssm.lds.base ssm.lds.dynamics ssm.lds.emissions ssm.lds.initial ssm.lds.models ssm.plots ssm.twarhmm.emissions ssm.twarhmm.models ssm.utils

N

NormalFactorialEmissions (class in ssm.factorial_hmm.emissions)

NormalFactorialHMM (class in ssm.factorial_hmm.models)

P

plot_dynamics_2d() (in module ssm.plots) PoissonEmissions (class in ssm.hmm.emissions) (class in ssm.lds.emissions)

PoissonHMM (class in ssm.hmm.models) PoissonLDS (class in ssm.lds.models)

R

random_rotation() (in module ssm.utils)

S

sample() (ssm.arhmm.base.AutoregressiveHMM method) (ssm.base.SSM method) SimpleStickyTransitions (class in ssm.hmm.transitions) SSM (class in ssm.base) ssm.arhmm.base module ssm.arhmm.emissions module ssm.arhmm.models module ssm.base module ssm.factorial_hmm.base module ssm.factorial_hmm.emissions module ssm.factorial_hmm.initial module ssm.factorial_hmm.models module ssm.factorial_hmm.transitions module ssm.hmm.base module ssm.hmm.emissions module ssm.hmm.initial module ssm.hmm.models module

ssm.hmm.transitions module ssm.inference.em module ssm.inference.laplace_em module ssm.lds.base module ssm.lds.dynamics module ssm.lds.emissions module ssm.lds.initial module ssm.lds.models module ssm.plots module ssm.twarhmm.emissions module ssm.twarhmm.models module ssm.utils module ssm_pbar() (in module ssm.utils) StandardInitialCondition (class in ssm.hmm.initial) (class in ssm.lds.initial) StationaryDynamics (class in ssm.lds.dynamics) StationaryTransitions (class in ssm.hmm.transitions)

T

test_and_find_inequality() (in module ssm.utils) TimeWarpedAutoregressiveEmissions (class in ssm.twarhmm.emissions) Transitions (class in ssm.hmm.transitions)

tree_all_equal() (in module ssm.utils) tree_concatenate() (in module ssm.utils) tree_get() (in module ssm.utils)

V

Verbosity (class in ssm.utils)