torch_mist.decomposition.tinfomax
Module Contents
Classes
Attributes
- torch_mist.decomposition.tinfomax.DEFAULT_MAX_ITERATIONS = 5000
- torch_mist.decomposition.tinfomax.DEFAULT_BATCH_SIZE = 64
- class torch_mist.decomposition.tinfomax.TInfoMax(n_dim: int, lagtime: int, normalize_inputs: bool = True, whiten: bool = False, proj: torch.nn.Module | None = None, model: torch_mist.nn.Model | None = None, proj_params: Dict[str, Any] | None = None, y_proj_params: Dict[str, Any] | None = None, model_params: Dict[str, Any] | None = None)
Bases:
torch_mist.decomposition.MID- _instantiate_model(*args, **kwargs)
- fit(X: numpy.ndarray | torch.Tensor, **train_params)