source: trunk/src/InputRanker.h @ 120

// $Id: InputRanker.h 120 2004-07-20 16:25:29Z peter $

#ifndef _theplu_cpptools_inputranker_ 
#define _theplu_cpptools_inputranker_ 

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

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


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

    ///
    /// Constructor taking data, target, weight, pointer to a Score
    /// object and vector defining what samples to use (default is to
    /// use all samples)
    ///
    InputRanker(const gslapi::matrix&, 
                const gslapi::vector&, 
                const gslapi::matrix&, 
                Score&, 
                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