mbi.estimation.DualAveraging

class mbi.estimation.DualAveraging(marginal_oracle: MarginalOracle | None = None, mesh: Mesh | None = None)[source]

Bases: Estimator

Regularized Dual Averaging estimator for graphical models.

RDA is an accelerated proximal algorithm for solving a smooth convex optimization problem over the marginal polytope. This algorithm requires knowledge of the Lipschitz constant of the gradient of the loss function.

marginal_oracle

The function to compute marginals from potentials. If None (default), uses default_oracle() to auto-select.

Type:: mbi.marginal_oracles.MarginalOracle | None

mesh

JAX sharding mesh.

Type:: jax._src.mesh.Mesh | None

Methods

`__init__`
`estimate`	Estimate a Model from noisy marginal measurements.
`precompile`	Warm up the JIT cache for `estimate` asynchronously.

Attributes

`marginal_oracle`
`mesh`

marginal_oracle: MarginalOracle | None = None

mesh: Mesh | None = None