InferTree.h

// Tree and DAG exact inference class interface
// Written by Sergey G. Kosov in 2015 for Project X
#pragma once

#include "MessagePassing.h"

namespace DirectGraphicalModels 
{
    // ==================== Tree Infer Class ==================
    class CInferTree : public CMessagePassing
    {
    public:
        DllExport CInferTree(CGraphPairwise & graph) : CMessagePassing(graph) {}
        DllExport virtual ~CInferTree(void) {}


    protected:
        DllExport virtual void calculateMessages(unsigned int nIt);
    };
}