source: trunk/lib/classifier/NCC.h @ 525

// $Id$

#ifndef _theplu_classifier_ncc_ 
#define _theplu_classifier_ncc_ 

#include <c++_tools/gslapi/matrix.h>

#include <c++_tools/classifier/SupervisedClassifier.h>

#include <map>

namespace theplu {

  namespace statistics {
    class Distance;
    class Score;
  }

namespace classifier {  

  class Target;
  class DataLookup1D;
  class DataLookup2D;

  ///
  /// Class for Nearest Centroid Classification
  ///

  class NCC : public SupervisedClassifier
  {
  
  public:
    ///
    /// Constructor taking the training data, the target vector and
    /// the distance measure as input. 
    ///
    NCC(const DataLookup2D&, const Target&, const statistics::Distance&);

    NCC(const DataLookup2D&, const Target&, const statistics::Distance&, 
        statistics::Score&, const size_t);

    virtual ~NCC();

    const gslapi::matrix& centroids(void) const {return centroids_;}

    inline SupervisedClassifier* 
    make_classifier(const DataLookup2D&, const Target&) const;
    
    bool train();


    ///
    /// Calculate the scores to each centroid for a test sample
    ///
    void predict(const DataLookup1D&, gslapi::vector&) const;
    
    ///
    /// Calculate the scores to each centroid for test samples
    ///
    void predict(const DataLookup2D&, gslapi::matrix&) const;


  private:
    gslapi::matrix centroids_;
    const statistics::Distance& distance_;                 
    const DataLookup2D& matrix_;

  };

  ///
  /// The output operator for the NCC class.
  ///
  std::ostream& operator<< (std::ostream&, const NCC&);
  
  
}} // of namespace classifier and namespace theplu

#endif