source: branches/0.4-stable/yat/classifier/CrossValidationSampler.h @ 1392

#ifndef _theplu_yat_classifier_crossvalidation_sampler_ 
#define _theplu_yat_classifier_crossvalidation_sampler_

// $Id: CrossValidationSampler.h 1392 2008-07-28 19:35:30Z peter $

/*
  Copyright (C) 2006 Jari Häkkinen, Peter Johansson, Markus Ringnér
  Copyright (C) 2007 Jari Häkkinen, 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 "Sampler.h"

namespace theplu {
namespace yat {
namespace classifier {

  class Target;

  ///
  /// @brief Class splitting a set into training set and validation
  /// set in a crossvalidation manner.
  /// 
  /// This is done in a balanced way, meaning
  /// the proportions between the classes in the trainingset is close
  /// to the proportions in the whole dataset. In the first \a k
  /// rounds each sample is returned k-1 times, for next round the
  /// samples are shuffled and... In total there are N partitions, in
  /// other words, each sample is in validation roughly N/k
  /// 
  class CrossValidationSampler : public Sampler
  {

  public:
    ///
    /// @brief Constructor 
    ///
    /// @a target targets.    
    /// @a N total number of partitions.
    /// @a k for k-fold crossvalidation.
    ///
    CrossValidationSampler(const Target& target, const size_t N,const size_t k);

    ///
    /// Destructor
    ///
    virtual ~CrossValidationSampler();

  private:
    void build(const Target& target, size_t N, size_t k); 
    const size_t k_;

  };

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

#endif