source: trunk/lib/classifier/EnsembleBuilder.h @ 520

// $Id: EnsembleBuilder.h 520 2006-02-22 15:01:10Z peter $

#ifndef _theplu_classifier_ensemblebuilder_ 
#define _theplu_classifier_ensemblebuilder_ 

#include <c++_tools/statistics/Averager.h>

#include <vector>

namespace theplu {
namespace classifier {  

  class CrossSplitter;
  class DataLookup2D;
  class SupervisedClassifier;

  ///
  /// Class for ensembles of supervised classifiers
  ///

  class EnsembleBuilder
  {
  
  public:
    ///
    /// Constructor.
    ///
    EnsembleBuilder(const SupervisedClassifier&, CrossSplitter&);

    ///
    /// Destructor.
    ///
    virtual ~EnsembleBuilder(void);


    ///
    /// Generate ensemble. Function trains each member of the Ensemble.
    ///
    void build(void);

    ///
    /// @Return classifier
    ///
    inline const SupervisedClassifier& classifier(size_t i) const
    {
      return *(classifier_[i]);
    }
      
    ///
    /// @Return Number of classifiers in ensemble
    ///
    inline u_long size(void) const
    {
      return classifier_.size();
    }

    ///
    /// Generate validation data for ensemble
    ///
    const std::vector<std::vector<statistics::Averager> >& validate(void);
    


  private:
  
    const SupervisedClassifier& mother_;
    CrossSplitter& cross_splitter_;
    std::vector<SupervisedClassifier*> classifier_;
    std::vector<std::vector<statistics::Averager> > validation_result_;

  };
  
}} // of namespace classifier and namespace theplu

#endif