NICE_VisLearning/regression/npregression/RegKNN.h

/**
* @file RegKNN.h
* @brief Implementation of k-Nearest-Neighbor algorithm for regression purposes
* @author Frank Prüfer
* @date 08/29/2013

*/ 
#ifndef REGKNNINCLUDE
#define REGKNNINCLUDE

#include "core/vector/VectorT.h"
#include "core/vector/VVector.h"
#include "core/vector/MatrixT.h"

#include "core/basics/Config.h"

#include <core/vector/Distance.h>

#include "vislearning/regression/regressionbase/RegressionAlgorithm.h"

namespace OBJREC
{ 
class RegKNN : public RegressionAlgorithm
{
  protected:
    int K;
    
    /** set of data points */
    NICE::VVector dataSet;
    
    /** set of responses according to dataset */
    NICE::Vector labelSet;
    
    /** used distance function */
    NICE::VectorDistance<double> *distancefunc;
  
  public:
    /** simple constructor */
    RegKNN( const NICE::Config *conf, NICE::VectorDistance<double> *distancefunc = NULL );
    
    /** simple destructor */
    virtual ~RegKNN();
    
    /** predict response using simple vector */
    double predict ( const NICE::Vector & x );
    
    /** teach whole set at once */
    void teach ( const NICE::VVector & dataSet, const NICE::Vector & labelSet );

//     /** teach one data point at a time */
//     void teach ( const NICE::Vector & x, const double & y );
};
}	//namespace

#endif