MAPInference.h

浏览该文件的文档.

/*
 * 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/lib/config.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,
    GRAPH_CUT = 4,
    GEMPLP = 5
};

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