mbi.estimation.MirrorDescentState
- class mbi.estimation.MirrorDescentState(potentials: CliqueVector, alpha: jax.Array | float, loss: jax.Array | float, mu: CliqueVector)[source]
Bases:
NamedTupleState for Algorithm 1 of https://arxiv.org/pdf/1901.09136.
Create new instance of MirrorDescentState(potentials, alpha, loss, mu)
Methods
__init__countReturn number of occurrences of value.
indexReturn first index of value.
Attributes
Alias for field number 1
Alias for field number 2
Alias for field number 3
Alias for field number 0
- potentials: CliqueVector
Alias for field number 0
- alpha: Array | float
Alias for field number 1
- loss: Array | float
Alias for field number 2
- mu: CliqueVector
Alias for field number 3