DTBStandard.h 922 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. /**
  2. * @file DTBStandard.h
  3. * @brief normal decision tree
  4. * @author Erik Rodner
  5. * @date 05/06/2008
  6. */
  7. #ifndef DTBSTANDARDINCLUDE
  8. #define DTBSTANDARDINCLUDE
  9. #include "core/vector/VectorT.h"
  10. #include "core/vector/MatrixT.h"
  11. #include "core/basics/Config.h"
  12. #include "DecisionTreeBuilder.h"
  13. namespace OBJREC {
  14. /** normal decision tree */
  15. class DTBStandard : public DecisionTreeBuilder
  16. {
  17. protected:
  18. double minimum_information_gain;
  19. double minExamples;
  20. DecisionNode *buildnode_allfeatures (
  21. const FeaturePool & parameters,
  22. FeatureStorage & featureStorage,
  23. const FullVector & example_counts );
  24. public:
  25. /** simple constructor */
  26. DTBStandard( const NICE::Config *conf, std::string section );
  27. /** simple destructor */
  28. virtual ~DTBStandard();
  29. DecisionNode *build ( const FeaturePool & fp,
  30. const Examples & examples, int maxClassNo );
  31. };
  32. } // namespace
  33. #endif