`torch_mist.estimators.generative.implementations.doe`

Module Contents

Classes

class torch_mist.estimators.generative.implementations.doe.DoE(q_Y_given_X: pyro.distributions.ConditionalDistribution, q_Y: torch.distributions.Distribution)

Bases: torch_mist.estimators.generative.base.ConditionalGenerativeMIEstimator

infomax_gradient: Dict[str, bool]

approx_log_p_y(y: torch.Tensor, x: torch.Tensor | None = None) → torch.Tensor

batch_loss(x: torch.Tensor, y: torch.Tensor) → torch.Tensor

__repr__()