pub struct MeanFieldInference {
pub max_iterations: usize,
pub tolerance: f64,
}Expand description
Mean-field variational inference.
Approximates the joint distribution with a product of independent marginals: Q(X₁, …, Xₙ) = ∏ᵢ Qᵢ(Xᵢ)
Fields§
§max_iterations: usizeMaximum iterations
tolerance: f64Convergence tolerance
Implementations§
Source§impl MeanFieldInference
impl MeanFieldInference
Sourcepub fn run(&self, graph: &FactorGraph) -> Result<HashMap<String, ArrayD<f64>>>
pub fn run(&self, graph: &FactorGraph) -> Result<HashMap<String, ArrayD<f64>>>
Run mean-field variational inference.
Returns approximate marginals for each variable.
Sourcepub fn compute_elbo(
&self,
graph: &FactorGraph,
q_distributions: &HashMap<String, ArrayD<f64>>,
) -> Result<f64>
pub fn compute_elbo( &self, graph: &FactorGraph, q_distributions: &HashMap<String, ArrayD<f64>>, ) -> Result<f64>
Compute ELBO (Evidence Lower BOund).
ELBO = E[log p(X, Z)] - E[log q(Z)]
Trait Implementations§
Auto Trait Implementations§
impl Freeze for MeanFieldInference
impl RefUnwindSafe for MeanFieldInference
impl Send for MeanFieldInference
impl Sync for MeanFieldInference
impl Unpin for MeanFieldInference
impl UnwindSafe for MeanFieldInference
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.