torch_mist.estimators.transformed.implementations

Submodules

• torch_mist.estimators.transformed.implementations.binned
• torch_mist.estimators.transformed.implementations.flip
• torch_mist.estimators.transformed.implementations.pq

Package Contents

Classes

BinnedMIEstimator
PQ

class torch_mist.estimators.transformed.implementations.BinnedMIEstimator(quantize_x: torch_mist.quantization.functions.QuantizationFunction | None = None, quantize_y: torch_mist.quantization.functions.QuantizationFunction | None = None, temperature: float = 1.0)

Bases: torch_mist.estimators.transformed.base.TransformedMIEstimator

lower_bound = True

class torch_mist.estimators.transformed.implementations.PQ(q_QY_given_X: pyro.distributions.ConditionalDistribution, quantize_y: torch_mist.quantization.functions.QuantizationFunction, temperature: float = 1.0)

Bases: torch_mist.estimators.transformed.base.TransformedMIEstimator