ITrain.h

// Interface class for random model training
// Written by Sergey G. Kosov in 2012 for Project X
#pragma once

#include "BaseRandomModel.h"

namespace DirectGraphicalModels
{
    // ================================ Train Class ================================
    class ITrain : public virtual CBaseRandomModel
    {
    public:
        DllExport ITrain(byte nStates, word nFeatures) : CBaseRandomModel(nStates), m_nFeatures(nFeatures) {}
        DllExport virtual ~ITrain(void) = default;

        DllExport virtual void  train(bool doClean = false) = 0;    
        DllExport word          getNumFeatures(void) const { return m_nFeatures; }


    private:
        word    m_nFeatures;                        
    };



}