mbi.estimation.LBFGS
- class mbi.estimation.LBFGS(marginal_oracle: MarginalOracle | None = None)[source]
Bases:
EstimatorL-BFGS estimator for graphical models.
Optimizes the potentials (theta) directly via L-BFGS, back-propagating through the marginal inference oracle. The loss is convex w.r.t. marginals but typically non-convex w.r.t. potentials; in practice, L-BFGS still converges well.
See “Learning Graphical Model Parameters with Approximate Marginal Inference”.
- marginal_oracle
The function to compute marginals from potentials. If
None(default), usesdefault_oracle()to auto-select.- Type:
Methods
__init__estimateEstimate a Model from noisy marginal measurements.
precompileWarm up the JIT cache for
estimateasynchronously.Attributes
- marginal_oracle: MarginalOracle | None = None