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

---

ComputeKernelData.h

Go to the documentation of this file.

#ifndef Impala_Core_ImageSet_ComputeKernelData_h
#define Impala_Core_ImageSet_ComputeKernelData_h

#include "Core/ImageSet/Reporter.h"
#include "Core/Training/PrecomputeTask.h"

namespace Impala
{
namespace Core
{
namespace ImageSet
{

class ComputeKernelData : public Listener
{
public:

    typedef Persistency::KernelMatrixLocator KernelMatrixLocator;

    ComputeKernelData(Reporter* reporter, CmdOptions& options)
    {
        if (options.GetNrArg() < 5)
        {
            ILOG_ERROR("Missing argument");
            return;
        }
        String name2 = options.GetArg(2);
        mDevelSet = Database::MakeRawDataSet(name2, true);
        mModel = options.GetArg(3);
        mKernelMatrixName = options.GetArg(4);
        mFeatureIndexCat = options.GetString("featureIndexCat");
        KernelMatrixLocator loc(mDevelSet->GetLocator(), true, "", "",
                                mModel, mKernelMatrixName, "");
        loc.SetFeatureIndexCat(mFeatureIndexCat);
        loc.SetStartNode(0);
        loc.SetNodeCount(Link::Mpi::NrProcs());
        loc.SetDoParts(0);
        mDA = Persistency::DistributedAccessRepository().Get(loc);
    }

    virtual
    ~ComputeKernelData()
    {
        delete mDA;
        delete mDevelSet;
    }

    virtual void
    HandleNewWalk(ImageSet* is, String walkType)
    {
        mWalkType = walkType;
    }

    virtual void
    HandleDoneDir(ImageSet* is, int dirId)
    {
        Training::PrecomputeTask::ComputeTestKernel
            (is->GetLocator(), mWalkType, is->GetContainer(dirId), mDevelSet,
             mDA, mFeatureIndexCat, mModel, mKernelMatrixName);
    }

private:

    Database::RawDataSet*      mDevelSet;
    String                     mWalkType;
    String                     mFeatureIndexCat;
    String                     mModel;
    String                     mKernelMatrixName;
    Matrix::DistributedAccess* mDA;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(ComputeKernelData, Impala.Core.ImageSet);

} // namespace ImageSet
} // namespace Core
} // namespace Impala

#endif

---