FPCDecisionTree.h 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. /**
  2. * @file FPCDecisionTree.h
  3. * @brief simple decision tree classifier
  4. * @author Erik Rodner
  5. * @date 04/21/2008
  6. */
  7. #ifndef FPCDECISIONTREEINCLUDE
  8. #define FPCDECISIONTREEINCLUDE
  9. #include "core/vector/VectorT.h"
  10. #include "core/vector/MatrixT.h"
  11. #include "core/basics/Config.h"
  12. #include "vislearning/classifier/classifierbase/FeaturePoolClassifier.h"
  13. #include "vislearning/cbaselib/CachedExample.h"
  14. #include "DecisionTreeBuilder.h"
  15. #include "DecisionTree.h"
  16. #undef ROADWORKS
  17. #define ROADWORKS fthrow( Exception, "Persistent interface not yet implemented.");
  18. namespace OBJREC {
  19. /** simple decision tree classifier */
  20. class FPCDecisionTree : public FeaturePoolClassifier
  21. {
  22. protected:
  23. const NICE::Config *conf;
  24. DecisionTree *dt;
  25. DecisionTreeBuilder *builder;
  26. public:
  27. /** simple constructor */
  28. FPCDecisionTree( const NICE::Config *conf, std::string section );
  29. /** simple destructor */
  30. virtual ~FPCDecisionTree();
  31. virtual ClassificationResult classify ( Example & pce );
  32. virtual void train ( FeaturePool & fp, Examples & examples );
  33. DecisionTree & getTree () { return *dt; };
  34. void restore (std::istream & is, int format = 0) { ROADWORKS };
  35. void store (std::ostream & os, int format = 0) const { ROADWORKS };
  36. void clear () { ROADWORKS };
  37. FeaturePoolClassifier *clone () const { ROADWORKS };
  38. };
  39. } // namespace
  40. #undef ROADWORKS
  41. #endif