source: trunk/test/knn_test.cc @ 916

// $Id: knn_test.cc 916 2007-09-30 00:50:10Z peter $

#include "yat/classifier/KNN.h"
#include "yat/classifier/MatrixLookupWeighted.h"
#include "yat/statistics/euclidean_vector_distance.h"
#include "yat/statistics/pearson_vector_distance.h"
#include "yat/utility/matrix.h"


#include <cassert>
#include <fstream>
#include <iostream>
#include <list>
#include <string>
#include <vector>


using namespace theplu::yat;

int main(const int argc,const char* argv[])

{  
  std::ostream* error;
  if (argc>1 && argv[1]==std::string("-v"))
    error = &std::cerr;
  else {
    error = new std::ofstream("/dev/null");
    if (argc>1)
      std::cout << "knn_test -v : for printing extra information\n";
  }
  *error << "testing knn" << std::endl;
  bool ok = true;
  
  std::ifstream is("data/sorlie_centroid_data.txt");
  utility::matrix data(is,'\t');
  is.close();
  
  is.open("data/sorlie_centroid_classes.txt");
  classifier::Target targets(is);
  is.close();

  // Generate weight matrix with 0 for missing values and 1 for others.
  utility::matrix weights(data.rows(),data.columns(),0.0);
  utility::nan(data,weights);

  classifier::MatrixLookupWeighted dataviewweighted(data,weights);
  classifier::KNN<statistics::pearson_vector_distance_tag> knn(dataviewweighted,targets);
  *error << "Training KNN" << std::endl;
  knn.train();
  
  utility::matrix prediction;
  knn.predict(dataviewweighted,prediction);
  
  if(!ok) {
    *error << "vector_distance_test failed" << std::endl;
  }
  if (error!=&std::cerr)
    delete error;
  if (ok=true)  
    return 0;
  return -1;
}