MachineEvaluation.h

Go to the documentation of this file.

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * Copyright (C) 2012 Jacob Walker
 *
 * Some code adapted from CrossValidation class by
 * Heiko Strathmann
 */

#ifndef CMACHINEEVALUATION_H_
#define CMACHINEEVALUATION_H_

#include <shogun/base/SGObject.h>
#include <shogun/evaluation/Evaluation.h>
#include <shogun/evaluation/EvaluationResult.h>
#include <shogun/evaluation/MachineEvaluation.h>

namespace shogun
{

class CMachine;
class CFeatures;
class CLabels;
class CSplittingStrategy;
class CEvaluation;

class CMachineEvaluation: public CSGObject
{

public:

    CMachineEvaluation();

    CMachineEvaluation(CMachine* machine, CFeatures* features, CLabels* labels,
            CSplittingStrategy* splitting_strategy,
            CEvaluation* evaluation_criterion, bool autolock = true);

    CMachineEvaluation(CMachine* machine, CLabels* labels,
            CSplittingStrategy* splitting_strategy,
            CEvaluation* evaluation_criterion, bool autolock = true);


    virtual ~CMachineEvaluation();

    EEvaluationDirection get_evaluation_direction();

    virtual CEvaluationResult* evaluate() = 0;

    CMachine* get_machine() const;

    void set_autolock(bool autolock) { m_autolock = autolock; }

protected:

    virtual void init();

protected:

    CMachine* m_machine;

    CFeatures* m_features;

    CLabels* m_labels;

    CSplittingStrategy* m_splitting_strategy;

    CEvaluation* m_evaluation_criterion;

    bool m_autolock;

    bool m_do_unlock;

};

} /* namespace shogun */

#endif /* CMACHINEEVALUATION_H_ */