torch_mist.decomposition

Submodules

Package Contents

Classes

MID

VIB

CEB

MIB

TIB

class torch_mist.decomposition.MID

Bases: torch_mist.decomposition.base.DimensionalityReduction

_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_proj(x_dim: int) → torch.nn.Module
_instantiate_y_proj(y_dim: int) → torch.nn.Module | None
_train_model(data: torch_mist.utils.data.utils.TensorDictLike, **train_params)
_instantiate_model(x_dim: int, y_dim: int)
class torch_mist.decomposition.VIB(*args, beta: float = 0.01, **kwargs)

Bases: torch_mist.decomposition.base.StochasticDimensionalityReduction

_add_default_mi_estimator_params(model_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]
_train_model(data: torch_mist.utils.data.utils.TensorDictLike, **train_params)
_instantiate_y_proj(y_dim: int) → torch.nn.Module | None
_instantiate_model(x_dim: int, y_dim: int)
class torch_mist.decomposition.CEB(*args, beta: float = 0.01, conditional_dist_params: Dict[str, Any] | None = None, **kwargs)

Bases: torch_mist.decomposition.base.StochasticDimensionalityReduction

_add_default_conditional_dist_params(cond_params: Dict[str, Any] | None) → Dict[str, Any]
_add_default_mi_estimator_params(model_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]
_train_model(data: torch_mist.utils.data.utils.TensorDictLike, **train_params)
_instantiate_y_proj(y_dim: int) → torch.nn.Module | None
_instantiate_model(x_dim: int, y_dim: int)
class torch_mist.decomposition.MIB(*args, beta: float = 0.01, **kwargs)

Bases: torch_mist.decomposition.base.StochasticDimensionalityReduction

_add_default_mi_estimator_params(model_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]
_add_default_y_proj_params(proj_params: Dict[str, Any] | None) → Dict[str, Any]
_instantiate_y_proj(y_dim: int) → torch.nn.Module
_train_model(data: torch_mist.utils.data.utils.TensorDictLike, **train_params)
_instantiate_model(x_dim: int, y_dim: int)
class torch_mist.decomposition.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)