source: trunk/test/consensus_inputranker_test.cc @ 668

// $Id: consensus_inputranker_test.cc 668 2006-10-06 07:05:50Z peter $

// C++ tools include
////////////////////
#include <c++_tools/classifier/ConsensusInputRanker.h>
#include <c++_tools/statistics/ROC.h>
#include <c++_tools/utility/matrix.h>
#include <c++_tools/classifier/MatrixLookup.h>
#include <c++_tools/classifier/CrossValidationSampler.h>
#include <c++_tools/classifier/IRRank.h>

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

using namespace std;

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

  ifstream is("data/rank_data.txt");
  theplu::utility::matrix data_tmp(is);
  theplu::classifier::MatrixLookup data(data_tmp);
  is.close();

  is.open("data/rank_target.txt");
  theplu::classifier::Target target(is);
  is.close();

  
  theplu::statistics::ROC roc;
  theplu::classifier::CrossValidationSampler sampler(target,30,3);
  *error << "Building Consensus_Inputranker" << std::endl;
  theplu::classifier::IRRank retrieve;
  theplu::classifier::ConsensusInputRanker cir(sampler,data,roc,retrieve);
  *error << "Done" << std::endl;

  if (cir.id(0)!=2 || cir.id(1)!=0 || cir.id(2)!=1){
    *error << "incorrect id" << endl;
    ok = false;
  }
  
  if (cir.rank(0)!=1 || cir.rank(1)!=2 || cir.rank(2)!=0){
    *error << "incorrect rank" << endl;
    ok=false;
  }

  theplu::utility::matrix flag(data.rows(),data.columns(),1);
  // Peter, fix weighted version instead
  theplu::classifier::ConsensusInputRanker cir2(sampler,data,roc,retrieve);

  if (cir2.id(0)!=2 || cir2.id(1)!=0 || cir2.id(2)!=1){
    *error << "incorrect id for weighted" << endl;
    ok=false;
  }
  
  if (cir2.rank(0)!=1 || cir2.rank(1)!=2 || cir2.rank(2)!=0){
    *error << "incorrect rank for weighted" << endl;
    ok=false;
  }
  
  
  if (error!=&std::cerr)
    delete error;

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