mbi.estimation.MirrorDescentState

class mbi.estimation.MirrorDescentState(potentials: CliqueVector, alpha: jax.Array | float, loss: jax.Array | float, mu: CliqueVector)[source]

Bases: NamedTuple

State for Algorithm 1 of https://arxiv.org/pdf/1901.09136.

Create new instance of MirrorDescentState(potentials, alpha, loss, mu)

Methods

__init__

count

Return number of occurrences of value.

index

Return first index of value.

Attributes

alpha

Alias for field number 1

loss

Alias for field number 2

mu

Alias for field number 3

potentials

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