source: trunk/yat/classifier/FeatureSelectorIR.cc @ 1437

// $Id$

/*
  Copyright (C) 2006 Jari Häkkinen, Peter Johansson
  Copyright (C) 2007 Jari Häkkinen, Peter Johansson, Markus Ringnér
  Copyright (C) 2008 Peter Johansson

  This file is part of the yat library, http://dev.thep.lu.se/yat

  The yat library is free software; you can redistribute it and/or
  modify it under the terms of the GNU General Public License as
  published by the Free Software Foundation; either version 2 of the
  License, or (at your option) any later version.

  The yat library is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  02111-1307, USA.
*/

#include "FeatureSelectorIR.h"
#include "FeatureSelector.h"
#include "MatrixLookup.h"
#include "MatrixLookupWeighted.h"
#include "InputRanker.h"
#include "Target.h"

#include <algorithm>
#include <cassert>

namespace theplu {
namespace yat {
namespace classifier {


  FeatureSelectorIR::FeatureSelectorIR(statistics::Score& score, size_t N, 
                                       size_t first)
    : FeatureSelector(N, first), score_(score)
  {
  }



  FeatureSelectorIR::FeatureSelectorIR(const FeatureSelectorIR& other)
    : FeatureSelector(other), score_(other.score_)
  {
  }



  FeatureSelectorIR& FeatureSelectorIR::operator=(const FeatureSelectorIR& other)
  {
    FeatureSelector::operator=(other);
    return *this;
  }


  void FeatureSelectorIR::update(const MatrixLookup& data, const Target& target)
  {
    assert(data.columns()==target.size());
    InputRanker ir = InputRanker(data, target, score_);
    std::vector<size_t>* features = new std::vector<size_t>(N_);
    std::copy(ir.id().begin()+first_, ir.id().begin()+first_+N_,
              features->begin());
    features_ = 
      utility::Index(utility::SmartPtr<const std::vector<size_t> >(features));
  }


  void FeatureSelectorIR::update(const MatrixLookupWeighted& data, 
                                 const Target& target)
  {
    assert(data.columns()==target.size());
    InputRanker ir = InputRanker(data, target, score_);
    std::vector<size_t>* features = new std::vector<size_t>(N_);
    std::copy(ir.id().begin()+first_, ir.id().begin()+first_+N_,
              features->begin());
    features_ = 
      utility::Index(utility::SmartPtr<const std::vector<size_t> >(features));
  }

}}} // of namespace classifier, yat, and theplu