`torch_mist.decomposition.tib`

Module Contents

Classes

Attributes

`DEFAULT_MAX_ITERATIONS`
`DEFAULT_BATCH_SIZE`

torch_mist.decomposition.tib.DEFAULT_MAX_ITERATIONS = 5000

torch_mist.decomposition.tib.DEFAULT_BATCH_SIZE = 64

class torch_mist.decomposition.tib.TIB(n_dim: int, lagtime: int, transition_params: Dict[str, Any] | None = None, *args, **kwargs)

Bases: torch_mist.decomposition.CEB

property transition

_add_default_conditional_dist_params(cond_params: Dict[str, Any] | None) → Dict[str, Any]

_add_default_model_params(model_params: Dict[str, Any] | None) → Dict[str, Any]

_add_default_proj_params(proj_params: Dict[str, Any] | None) → Dict[str, Any]

_instantiate_model(*args, **kwargs) → torch_mist.models.bottleneck.TIB

fit(X: numpy.ndarray | torch.Tensor, **train_params)