Struct JunctionTree 

pub struct JunctionTree { /* private fields */ }

A calibrated (or calibratable) junction tree for exact inference.

Implementations

impl JunctionTree

pub fn build( cfg: &JunctionTreeConfig, factors: &[(Vec<usize>, Vec<f64>)], ) -> SeqResult<Self>

Build a junction tree from a list of factors.

Each factor is (vars, table) where vars is the (unsorted is allowed) scope and table is a linear-domain non-negative potential table, row-major over the factor’s joint configuration in the given vars order. Tables are converted to the log domain internally.

pub fn from_mrf(mrf: &Mrf) -> SeqResult<Self>

Build a junction tree directly from a pairwise Mrf. Unary terms become single-variable factors and each pairwise term a two-variable factor. The Mrf stores energies (probability ∝ exp(-energy)), so tables are exp(-unary) and exp(-pairwise) respectively.

pub fn calibrate(&mut self) -> SeqResult<()>

Run Hugin two-pass message passing (collect to root, distribute from root) so that every clique stores its joint marginal (scaled by Z) and every separator is consistent with both incident cliques.

pub fn marginal(&self, var: usize) -> SeqResult<Vec<f64>>

Marginal distribution over a single variable after calibration, normalised to sum to 1.

pub fn clique_marginal(&self, clique_idx: usize) -> SeqResult<Vec<f64>>

Joint marginal over the variables of clique clique_idx, normalised to sum to 1. The output is row-major over the clique’s (sorted) variables.

pub fn log_partition(&self) -> SeqResult<f64>

Log partition function log Z (log of the normalisation constant).

After calibration every clique sums (in the linear domain) to Z, so we take the log-sum-exp of any clique’s log-potential. Before calibration this is generally not Z; callers should calibrate first.

pub fn n_cliques(&self) -> usize

Number of cliques in the tree.

pub fn n_separators(&self) -> usize

Number of separators (tree edges) in the tree.

pub fn cliques(&self) -> &[Clique]

Read-only view of the cliques.

pub fn separator_vars(&self, sep_idx: usize) -> SeqResult<&[usize]>

Variables of separator sep_idx (sorted); useful for tests checking the running-intersection property.

pub fn separator_cliques(&self, sep_idx: usize) -> SeqResult<(usize, usize)>

The (clique_a, clique_b) incident to separator sep_idx.

Trait Implementations

impl Clone for JunctionTree

fn clone(&self) -> JunctionTree

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for JunctionTree

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.