torch_mist.estimators.transformed.factories

Module Contents

Functions

binned(...)
pq(→ torch_mist.estimators.transformed.implementations.PQ)

torch_mist.estimators.transformed.factories.binned(quantize_x: torch_mist.quantization.QuantizationFunction | str | None = 'kmeans', quantize_y: torch_mist.quantization.QuantizationFunction | str | None = 'kmeans', temperature: float = 0.1, n_bins: int | None = 32, x_dim: int | None = None, y_dim: int | None = None, **kwargs) → torch_mist.estimators.transformed.implementations.BinnedMIEstimator

torch_mist.estimators.transformed.factories.pq(quantize_y: torch_mist.quantization.QuantizationFunction | str | None = 'kmeans', x_dim: int | None = None, hidden_dims: List[int] | None = None, q_QY_given_X: pyro.distributions.ConditionalDistribution | None = None, temperature: float = 0.1, n_bins: int | None = 32, y_dim: int | None = None, **kwargs) → torch_mist.estimators.transformed.implementations.PQ