`torch_mist.decomposition.mid`

Module Contents

Classes

Attributes

`DEFAULT_MAX_ITERATIONS`
`DEFAULT_BATCH_SIZE`

torch_mist.decomposition.mid.DEFAULT_MAX_ITERATIONS = 5000

torch_mist.decomposition.mid.DEFAULT_BATCH_SIZE = 64

class torch_mist.decomposition.mid.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)