hbayes-0.2: Inference with Discrete Bayesian Networks

Safe HaskellSafe-Infered

Bayes.FactorElimination

Contents

Description

Algorithms for factor elimination

Synopsis

Moral graph

moralGraph :: (NamedGraph g, FoldableWithVertex g, DirectedGraph g) => g () b -> UndirectedSG () bSource

For the junction tree construction, only the vertices are needed during the intermediate steps. So, the moral graph is returned without any vertex data.

Triangulation

nodeComparisonForTriangulation :: (UndirectedGraph g, Factor f) => g a f -> Vertex -> Vertex -> OrderingSource

Node selection comparison function used for triangulating the graph

numberOfAddedEdges :: UndirectedGraph g => g a b -> Vertex -> IntSource

Number of edges added when connecting all neighbors

weight :: (UndirectedGraph g, Factor f) => g a f -> Vertex -> IntSource

Weight of a node

triangulateSource

Arguments

:: Graph g 
=> (Vertex -> Vertex -> Ordering)

Criterion function for triangulation

-> g () b 
-> ([VertexCluster], g () b)

Returns the clusters and the triangulated graph

Triangulate a graph using a cost function The result is the triangulated graph and the list of clusters which may not be maximal.

Junction tree

createClusterGraph :: (UndirectedGraph g, Factor f, Graph g') => g' e f -> [VertexCluster] -> g Int ClusterSource

Create the cluster graph

data Cluster Source

Cluster of discrete variables. Discrete variables instead of vertices are needed because the factor are using DV and we need to find which factors must be contained in a given cluster.

Instances

Eq Cluster 
Ord Cluster 
Show Cluster 
IsCluster Cluster 
Factor f => Message f Cluster 

createJunctionTreeSource

Arguments

:: (DirectedGraph g, FoldableWithVertex g, NamedGraph g, Factor f, Show f) 
=> (UndirectedSG () f -> Vertex -> Vertex -> Ordering)

Weight function on the moral graph

-> BayesianNetwork g f

Input directed graph

-> JunctionTree f

Junction tree

Create a function tree

createUninitializedJunctionTreeSource

Arguments

:: (DirectedGraph g, FoldableWithVertex g, NamedGraph g, Factor f) 
=> (UndirectedSG () f -> Vertex -> Vertex -> Ordering)

Weight function on the moral graph

-> g () f

Input directed graph

-> JunctionTree f

Junction tree

Create a junction tree with only the clusters and no factors

type JunctionTree f = JTree Cluster fSource

Shenoy-Shafer message passing

collect :: (Ord c, Message a c) => JTree c a -> JTree c aSource

Collect message taking into account that the tree depth may be different for different leaves.

distribute :: (Ord c, Message a c) => JTree c a -> JTree c aSource

posterior :: Factor f => JunctionTree f -> DV -> Maybe fSource

Compute the marginal posterior (if some evidence is set on the junction tree) otherwise compute just the marginal prior.

Evidence

changeEvidenceSource

Arguments

:: (IsCluster c, Ord c, Factor f, Message f c) 
=> [DVI Int]

Evidence

-> JTree c f 
-> JTree c f 

Change evidence in the network

Test

data VertexCluster Source

A cluster containing only the vertices and not yet the factors

For debug

maximumSpanningTree :: (UndirectedGraph g, IsCluster c, Factor f, Ord c) => g Int c -> JTree c fSource

Implementing the Prim's algorithm for minimum spanning tree