MAPInference.h

Go to the documentation of this file.

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * Written (W) 2013 Shell Hu 
 * Copyright (C) 2013 Shell Hu 
 */

#ifndef __MAP_INFERENCE_H__
#define __MAP_INFERENCE_H__

#include <shogun/base/SGObject.h>
#include <shogun/lib/SGVector.h>
#include <shogun/structure/FactorGraph.h>
#include <shogun/labels/FactorGraphLabels.h>

namespace shogun
{

enum EMAPInferType
{
    TREE_MAX_PROD = 0,
    LOOPY_MAX_PROD = 1,
    LP_RELAXATION = 2,
    TRWS_MAX_PROD = 3,
    ITER_COND_MODE = 4,
    NAIVE_MEAN_FIELD = 5,
    STRUCT_MEAN_FIELD = 6,
};

class CMAPInferImpl;

class CMAPInference : public CSGObject
{
public:
    CMAPInference();

    CMAPInference(CFactorGraph* fg, EMAPInferType inference_method);

    virtual ~CMAPInference();

    virtual const char* get_name() const { return "MAPInference"; }

    virtual void inference();

    CFactorGraphObservation* get_structured_outputs() const;

    float64_t get_energy() const;

private:
    void init();

protected:
    CFactorGraph* m_fg; 

    CFactorGraphObservation* m_outputs;

    float64_t m_energy;

    CMAPInferImpl* m_infer_impl; 
};

class CMAPInferImpl : public CSGObject
{
public:
    CMAPInferImpl();

    CMAPInferImpl(CFactorGraph* fg);

    virtual ~CMAPInferImpl();

    virtual const char* get_name() const { return "MAPInferImpl"; }

    virtual float64_t inference(SGVector<int32_t> assignment) = 0;

private:
    void register_parameters();

protected:
    CFactorGraph* m_fg;
};

}

#endif