|| || || || || || || ||

---

TrainDataSrcFeatureTable.h

Go to the documentation of this file.

#ifndef Impala_Core_Training_TrainDataSrcFeatureTable_h
#define Impala_Core_Training_TrainDataSrcFeatureTable_h

#include "Core/Training/TrainDataSrc.h"
#include "Core/Training/SvmProblemBuilder.h"
#include "Core/Feature/FeatureTable.h"
#include "Link/Svm/LinkSvm.h"

namespace Impala
{
namespace Core
{
namespace Training
{

class TrainDataSrcFeatureTable : public TrainDataSrc
{
public:
    typedef Core::Vector::VectorTem<Real64> VectorReal64;

    TrainDataSrcFeatureTable(Table::AnnotationTable* annotation,
                             Feature::FeatureTable* features)
        : TrainDataSrc(annotation)
    {
        mFeatures = features;
        mVectorLength = mFeatures->GetFeatureVectorLength();
    }

    virtual
    ~TrainDataSrcFeatureTable()
    {
    }

    virtual svm_problem*
    MakeSvmProblem()
    {
        SvmProblemBuilder pb(mAnnotation, mSelection);
        pb.AddFeatureTable(mFeatures);
        return pb.MakeProblem();
    }

    virtual svm_problem*
    MakeSvmProblem(int i)
    {
        if(i >= mFeatures->Size())
        {
            ILOG_ERROR("index > mFeatures->Size: " << i);
            return MakeEmptyProblem();
        }
        Quid q = mFeatures->Get1(i);
        VectorReal64 vector = mFeatures->Get2(i);
        SvmProblemBuilder pb(mAnnotation, mSelection);
        svm_problem* ret = pb.MakeProblem(&vector);
        return ret;
    }

    // copy from file TrainDataSrcFeature...
    virtual void
    FreeProblem(svm_problem* p)
    {
        delete[] p->y;
        for(int i=0 ; i<p->l ; ++i)
            delete[] p->x[i];
        delete[] p->x;
        delete p;
    }

    virtual int
    GetVectorLength()
    {
        return mVectorLength;
    }

    virtual int
    Size()
    {
        ILOG_DEBUG("correct Size called");
        return mFeatures->Size();
    }

    virtual Quid
    GetQuid(int i)
    {
        ILOG_DEBUG("correct GetQuid called");
        return mFeatures->Get1(i);
    }

private:
    Feature::FeatureTable* mFeatures;
    int mVectorLength;

    ILOG_VAR_DECL;
};

ILOG_VAR_INIT(TrainDataSrcFeatureTable, Impala.Core.Training);

}//namespace Training
}//namespace Core
}//namespace Impala


#endif

---