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

---

ApplyConceptsHelperFeatures.h

Go to the documentation of this file.

#ifndef Impala_Core_Training_ApplyConceptsHelperFeatures_h
#define Impala_Core_Training_ApplyConceptsHelperFeatures_h

#include "Core/Training/ApplyConceptsHelper.h"
#include "Core/Table/Read.h"
#include "Core/Training/TrainDataSrcFeatureTable.h"
#include "Core/Training/KernelFunctions.h"

namespace Impala
{
namespace Core
{
namespace Training
{

class ApplyConceptsHelperFeatures : public ApplyConceptsHelper
{
public:
    ApplyConceptsHelperFeatures()
        : mInitialized(false), mFeatures(0)
    {
    }
    
    void
    Initialise(Database::RawDataSet* annoSet, String featureName)
    {
        mAnnoSet = annoSet;
        mFeatures = 0;
        mFeatureDef = Feature::FeatureDefinition(featureName);
        mInitialized = true;
    }
        
    String
    GetModelName()
    {
        return mFeatureDef.AsString();
    }

    FeatureTable*
    LoadFeatures(Database::PathCreator* pathCreator)
    {
        if(!mInitialized)
            ILOG_ERROR("not initialized");
        if(mFeatures)
            ILOG_ERROR("programming error");
        String path = pathCreator->GetFilePathFeatureData(false, false);
        mFeatures = FeatureTable::MakeFromFile(mFeatureDef, path,
                                               pathCreator->GetDatabase());
        return mFeatures;
    }

    Table::ScoreTable*
    Predict(String modelName)
    {
        if(!mInitialized)
            ILOG_ERROR("not initialized");
        Training::Classifier* classifier =
            Training::LoadClassifier(modelName, mAnnoSet->GetDatabase());
        TrainDataSrcFeatureTable dataSrc(0, mFeatures);
        ILOG_DEBUG("data size = " << dataSrc.Size());
        Table::ScoreTable* scores = classifier->Predict(&dataSrc);
        delete classifier;
        return scores;
    }

    void
    UnloadFeatures()
    {
        delete mFeatures;
        mFeatures = 0;
    }

private:
    Database::RawDataSet* mAnnoSet;
    FeatureTable* mFeatures;
    Feature::FeatureDefinition mFeatureDef;
    bool mInitialized;
    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(ApplyConceptsHelperFeatures, Impala.Core.Training);

} //namespace
} //namespace
} //namespace

#endif

---