TrainNodeGM.h

// Gaussian Model training class interface
// Written by Sergey G. Kosov in 2012 for Project X
#pragma once

#include "TrainNodeGMM.h"

namespace DirectGraphicalModels
{
    class CNDGauss;
    // ======================== Gaussian Model Train Class =========================
    class CTrainNodeGM : public CTrainNodeGMM
    {
    public:
        DllExport CTrainNodeGM(byte nStates, word nFeatures) : CBaseRandomModel(nStates), CTrainNodeGMM(nStates, nFeatures, 1) {}
        DllExport ~CTrainNodeGM(void) {}

        DllExport Mat getNodePotentials(const Mat &featureVector, float weight = 1.0f) {return CTrainNodeGMM::getNodePotentials(featureVector, weight);}    // This is done for doxygen comment
    };
}