TrainLink.h

// Base abstract class for random model links (inter-layer edges) training
// Written by Sergey G. Kosov in 2016 for Project X
#pragma once

#include "ITrain.h"

namespace DirectGraphicalModels
{
    // ============================= Link Train Class =============================
    class CTrainLink : public ITrain
    {
    public:
        DllExport CTrainLink(byte nStatesBase, byte nStatesOccl, word nFeatures) 
            : CBaseRandomModel(nStatesBase * nStatesOccl)
            , ITrain(nStatesBase * nStatesOccl, nFeatures)
            , m_nStatesBase(nStatesBase)
            , m_nStatesOccl(nStatesOccl)
        {}
        DllExport virtual ~CTrainLink(void) = default;

        
        DllExport void          addFeatureVec(const Mat &featureVectors, const Mat &gtb, const Mat &gto);
        DllExport void          addFeatureVec(const vec_mat_t &featureVectors, const Mat &gtb, const Mat &gto);
        DllExport virtual void  addFeatureVec(const Mat &featureVector, byte gtb, byte gto) = 0;
        DllExport virtual void  train(bool doClean = false) {}
        DllExport Mat           getLinkPotentials(const Mat &featureVector, float weight = 1.0f) const;


    protected:
        DllExport virtual Mat   calculateLinkPotentials(const Mat &featureVector) const = 0;


    protected:
        byte m_nStatesBase;     
        byte m_nStatesOccl;     
    };
}