source: trunk/test/crossvalidation_test.cc @ 453

// $Id: crossvalidation_test.cc 453 2005-12-15 21:49:23Z peter $

#include <c++_tools/classifier/CrossValidation.h>
#include <c++_tools/gslapi/vector.h>

#include <vector>
#include <cstdlib>
#include <iostream>

using namespace theplu;

int main()
{  

  gslapi::vector target(10,1);
  for (size_t i=0; i<5; i++) 
    target(i)=-1;

  classifier::CrossValidation cv(target,3);


  std::vector<size_t> training_set;
  std::vector<size_t> count(10);

  training_set = cv.next();
  for (unsigned int i=0; i<training_set.size(); i++) 
    count[training_set[i]]++;
  
  training_set = cv.next();
  for (unsigned int i=0; i<training_set.size(); i++) 
    count[training_set[i]]++;

  training_set = cv.next();
  for (unsigned int i=0; i<training_set.size(); i++) 
    count[training_set[i]]++;

  for (unsigned int i=0; i<10 ; i++) 
    if (count[i]!=2)
      return -1;

  return 0;
}