NICE_VisLearning/classifier/fpclassifier/randomforest/DTBClusterRandom.h

/** 
* @file DTBClusterRandom.h
* @brief build a decision tree for clustering
* @author Erik Rodner
* @date 05/01/2010
*/
#ifndef DTBCLUSTERRANDOMINCLUDE
#define DTBCLUSTERRANDOMINCLUDE

#ifdef NOVISUAL
#include <vislearning/nice_nonvis.h>
#else
#include <vislearning/nice.h>
#endif

#include "core/basics/Config.h"
#include "DecisionTreeBuilder.h"
#include "vislearning/cbaselib/CachedExample.h"


namespace OBJREC {

/** build a decision tree for clustering */
class DTBClusterRandom : public DecisionTreeBuilder
{
    protected:
		int max_depth;
		int min_examples;

		DecisionNode *buildRecursive ( const FeaturePool & fp, 
				 const Examples & examples,
				 std::vector<int> & examples_selection,
				 FullVector & distribution,
				 int maxClassNo,
				 int depth );

    public:
  
		/** simple constructor */
		DTBClusterRandom( const NICE::Config *conf, std::string section = "DTBClusterRandom" );
		  
		/** simple destructor */
		virtual ~DTBClusterRandom();
	 
		DecisionNode *build ( const FeaturePool & fp, 
				 const Examples & examples,
				 int maxClassNo );
    
};


} // namespace

#endif