source: trunk/src/InputRanker.h @ 89

// $Id: InputRanker.h 89 2004-05-28 17:44:42Z peter $

#ifndef _theplu_cpptools_inputranker_ 
#define _theplu_cpptools_inputranker_ 

// C++ tools include
/////////////////////
#include "vector.h"
#include "matrix.h"

// Standard C++ includes
////////////////////////


namespace theplu {
namespace cpptools {  
  ///
  /// Class for ranking the columns in a matrix, using ROC (this could
  /// and the target vector.
  ///   
  class InputRanker
  {
  
  public:
    ///
    /// Constructor taking data, target and vector defining what
    /// samples to use (default is to use all samples)
    ///
    InputRanker(const gslapi::matrix&, const gslapi::vector&, 
           const std::vector<size_t> = std::vector<size_t>());

    ///
    /// @return id of input ranked as number i
    ///
    inline size_t id(const size_t i) const {return id_[i];}
    
    ///
    /// @return rank for id i
    ///
    inline size_t rank(const size_t i) const {return rank_[i];}


  private:
    std::vector<size_t> train_set_;
    std::vector<size_t> id_;
    std::vector<size_t> rank_;

         
  };

}} // of namespace cpptools and namespace theplu

#endif