source: trunk/c++_tools/classifier/NBC.cc @ 662

// $Id: NBC.cc 662 2006-09-27 10:16:17Z peter $

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

#include <c++_tools/classifier/DataLookup2D.h>
#include <c++_tools/classifier/MatrixLookup.h>
#include <c++_tools/classifier/MatrixLookupWeighted.h>
#include <c++_tools/classifier/Target.h>
#include <c++_tools/statistics/AveragerWeighted.h>
#include <c++_tools/utility/matrix.h>

#include <vector>

namespace theplu {
namespace classifier {

  NBC::NBC(const MatrixLookup& data, const Target& target) 
    : SupervisedClassifier(target), data_(data)
  {
  }

  NBC::NBC(const MatrixLookupWeighted& data, const Target& target) 
    : SupervisedClassifier(target), data_(data)
  {
  }

  NBC::~NBC()   
  {
  }


  SupervisedClassifier* 
  NBC::make_classifier(const DataLookup2D& data, const Target& target) const 
  {     
    NBC* ncc=0;
    if(data.weighted()) {
      ncc=new NBC(dynamic_cast<const MatrixLookupWeighted&>(data),target);
    }
    else {
      ncc=new NBC(dynamic_cast<const MatrixLookup&>(data),target);
    }
    return ncc;
  }


  bool NBC::train()
  {   
    sigma_=centroids_=utility::matrix(data_.rows(), target_.nof_classes());
    utility::matrix nof_in_class(data_.rows(), target_.nof_classes());
    
    
    for(size_t i=0; i<data_.rows(); ++i) {
      std::vector<statistics::AveragerWeighted> aver(target_.nof_classes());
      for(size_t j=0; j<data_.columns(); ++j) {
        if (data_.weighted()){
          const MatrixLookupWeighted& data = 
            dynamic_cast<const MatrixLookupWeighted&>(data_);
            aver[target_(j)].add(data.data(i,j), data.weight(i,j));
        }
        else
          aver[target_(j)].add(data_(i,j),1.0);
      }
      for (size_t j=0; target_.nof_classes(); ++j){
        centroids_(i,j) = aver[j].mean();
        sigma_(i,j) = aver[j].variance();
      }
    }   
    trained_=true;
    return trained_;
  }


  void NBC::predict(const DataLookup2D& input,                    
                    utility::matrix& prediction) const
  {   
    std::cerr << "NBC::predict not implemented\n";
    exit(1);
  }


}} // of namespace classifier and namespace theplu