Crate tensorlogic_quantrs_hooks 

TL <-> QuantrS2 hooks (PGM/message passing as reductions).

This crate provides integration between TensorLogic and probabilistic graphical models (PGMs). It maps belief propagation and other message passing algorithms onto einsum reduction patterns.

Core Concepts

• Factor Graphs: Convert TLExpr predicates into factors
• Message Passing: Sum-product and max-product algorithms as tensor operations
• Inference: Marginalization and conditional queries via reductions

Architecture

TLExpr → FactorGraph → MessagePassing → Marginals
   ↓         ↓              ↓              ↓
Predicates Factors    Einsum Ops    Probabilities

Re-exports

pub use parameter_learning::BaumWelchLearner;

pub use parameter_learning::BayesianEstimator;

pub use parameter_learning::MaximumLikelihoodEstimator;

pub use parameter_learning::SimpleHMM;

pub use quantrs_hooks::DistributionExport;

pub use quantrs_hooks::DistributionMetadata;

pub use quantrs_hooks::ModelExport;

pub use quantrs_hooks::ModelStatistics;

pub use quantrs_hooks::QuantRSAssignment;

pub use quantrs_hooks::QuantRSDistribution;

pub use quantrs_hooks::QuantRSInferenceQuery;

pub use quantrs_hooks::QuantRSModelExport;

pub use quantrs_hooks::QuantRSParameterLearning;

pub use quantrs_hooks::QuantRSSamplingHook;

Modules

parameter_learning

Parameter learning algorithms for probabilistic graphical models.

quantrs_hooks

QuantRS2 integration hooks for probabilistic graphical models.

Structs

BayesianNetwork

Bayesian Network builder.

BetheApproximation

Bethe approximation for structured variational inference.

Clique

A clique in the junction tree (a maximal set of connected variables).

ConditionalQuery

Query for conditional probability P(X | Y = y).

ConditionalRandomField

Conditional Random Field builder (discriminative model for structured prediction).

ConvergenceStats

Convergence statistics for belief propagation.

EmissionFeature

Emission feature: fires when a specific label is paired with a specific observation.

ExpectationPropagation

Expectation Propagation algorithm for approximate inference.

Factor

A factor in a probabilistic graphical model.

FactorGraph

Factor graph representation for PGM.

FactorNode

Factor node in a factor graph.

GaussianEP

Gaussian EP for continuous variables with moment matching.

GaussianSite

Gaussian site approximation for continuous variables.

GibbsSampler

Gibbs sampling for approximate inference.

HiddenMarkovModel

Hidden Markov Model builder.

IdentityFeature

Simple identity feature: always returns 1.0

InferenceEngine

Inference engine for PGM queries.

JunctionTree

Junction tree structure for exact inference.

JunctionTreeEdge

Edge in the junction tree connecting two cliques.

LinearChainCRF

Linear-chain CRF for sequence labeling.

MarginalizationQuery

Query for marginal probability P(X).

MarkovRandomField

Markov Random Field builder (undirected graphical model).

MaxProductAlgorithm

Max-product algorithm (MAP inference).

MeanFieldInference

Mean-field variational inference.

Separator

A separator between two cliques (their shared variables).

Site

Site approximation for a single factor.

SumProductAlgorithm

Sum-product algorithm (belief propagation).

TransitionFeature

Transition feature: fires when transitioning from one state to another.

TreeReweightedBP

Tree-reweighted belief propagation (TRW-BP).

VariableElimination

Variable elimination algorithm for exact inference.

VariableNode

Variable node in a factor graph.

Enums

FactorOp

Operations on factors.

PgmError

Errors that can occur in PGM operations.

Traits

FeatureFunction

Feature function for linear-chain CRF.

MessagePassingAlgorithm

Trait for message passing algorithms.

Functions

condition

Compute conditional probability P(X | Y = y).

expr_to_factor_graph

Convert a TensorLogic expression to a factor graph.

marginalize

Compute marginal probability for a variable.

message_passing_reduce

Perform message passing inference on a factor graph.

Type Aliases

Assignment

Assignment of values to variables.

Result

Result type for PGM operations.