source: branches/peters_vector/lib/classifier/NCC.h @ 469

// $Id$

#ifndef _theplu_classifier_ncc_ 
#define _theplu_classifier_ncc_ 

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

#include <map>

namespace theplu {
namespace gslapi {
  class vector;
}
namespace statistics {
  class Score;
}

namespace classifier {  

  class DataView;
  class Score;
  class Target;
  class VectorView;

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

  class NCC
  {
  
  public:
    ///
    /// Default constructor (not implemented)
    ///
    NCC();

    ///
    /// Constructor taking the training data and the target vector as
    /// input. Performs the training of the NCC.
    ///
    NCC(const DataView&, const Target&);

    ///
    /// @todo Copy constructor.
    ///
    NCC(const NCC&);

    ///
    /// @todo The istream constructor.
    ///
    NCC(std::istream&);
         
    const gslapi::matrix& centroids(void) const {return centroids_;}

    const std::map<double,u_int>& classes(void) const {return classes_;}
    

    ///
    /// Calculate the scores to each centroid for a test sample
    ///
    void predict(const gslapi::vector&, statistics::Score&, 
                 gslapi::vector&);
    
  private:
    gslapi::matrix centroids_;
    std::map<double,u_int> classes_;
  };

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

#endif