source: trunk/test/kernel_test.cc @ 446

// $Id: kernel_test.cc 446 2005-12-15 17:11:29Z peter $


// C++ tools include
////////////////////
#include <c++_tools/gslapi/matrix.h>
#include <c++_tools/gslapi/vector.h>
#include <c++_tools/svm/KernelFunction.h>
#include <c++_tools/svm/PolynomialKernelFunction.h>
#include <c++_tools/svm/GaussianKernelFunction.h>
#include <c++_tools/svm/KernelView.h>
#include <c++_tools/svm/Kernel_MEV.h>
#include <c++_tools/svm/Kernel_SEV.h>

#include <cmath>
#include <cstdlib>
#include <fstream>
#include <iostream>
#include <vector>

using namespace theplu;

bool test_MEV(const gslapi::matrix& data, const svm::KernelFunction* kf, 
              const gslapi::matrix& control, const double error_bound)
{
  svm::Kernel_MEV kernel(data,*kf);
  for(u_int i=0;i<control.rows();i++)
    for(u_int j=0;j<control.columns();j++)
      if (fabs(kernel(i,j)-control(i,j))>error_bound)
        return false;

  // checking view
  std::vector<size_t> index(3);
  index[0]=1;
  index[1]=2;
  index[2]=3;
  svm::KernelView(kernel,index);

  return true;
}

bool test_SEV(const gslapi::matrix& data, const svm::KernelFunction* kf, 
              const gslapi::matrix& control, const double error_bound)
{
  svm::Kernel_SEV kernel(data,*kf);
  for(u_int i=0;i<control.rows();i++)
    for(u_int j=0;j<control.columns();j++)
      if (fabs(kernel(i,j)-control(i,j))>error_bound)
        return false;

  // checking view
  std::vector<size_t> index(3);
  index[0]=1;
  index[1]=2;
  index[2]=3;
  svm::KernelView(kernel,index);
  return true;
}


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 << "kernel_test -v : for printing extra information\n";
  }
  *error << "testing kernels" << std::endl;
  bool ok = true;

  // Peter, the hardcoded number below should be changed.
  double error_bound = 1e-8; 
  std::ifstream is("data/nm_data_centralized.txt");
  gslapi::matrix transposed_data(is);
  is.close();
  // Because how the kernel is treated is changed, data must be transposed.
  gslapi::matrix data=transposed_data;

  is.open("data/nm_kernel.txt");
  gslapi::matrix kernel_matlab(is);
  is.close();
  svm::KernelFunction* kf = new svm::PolynomialKernelFunction(); 
  ok = (ok && test_MEV(data,kf,kernel_matlab,error_bound)
        & test_SEV(data,kf,kernel_matlab,error_bound));
  delete kf;
  
  is.open("data/nm_kernel2.txt");
  gslapi::matrix kernel_matlab2(is);
  is.close();
  kf = new svm::PolynomialKernelFunction(2); 
  ok = (ok && test_MEV(data,kf,kernel_matlab2,error_bound)
        & test_SEV(data,kf,kernel_matlab2,error_bound));
  delete kf;

  // Checking that a GaussianKernelFunction object can be built at
  // compile time.
  kf = new svm::GaussianKernelFunction(); 
  delete kf;

  if (error!=&std::cerr)
    delete error;

  if (ok=true)  
    return 0;
  return -1;
}