123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104 |
- #ifndef DTESTIMATEAPRIORIINCLUDE
- #define DTESTIMATEAPRIORIINCLUDE
- #include "core/vector/VectorT.h"
- #include "core/vector/MatrixT.h"
-
- #include "core/basics/Config.h"
- #include "vislearning/optimization/mapestimation/MAPEstimation.h"
- #include "DecisionTree.h"
- namespace OBJREC {
- class DTEstimateAPriori
- {
- private:
- MAPEstimation *map_estimator;
- public:
-
-
- void calcPosteriori ( DecisionTree & tree,
- const FullVector & aprioriOld,
- const std::map<DecisionNode *, FullVector> & nodeprobOld,
- std::map<DecisionNode *, double> & nodeprobNew,
- const std::set<int> & supportClassNo,
- int newClassNo,
- std::map<DecisionNode *, FullVector> & posterioriResult );
-
- void calculateNodeProbabilitiesRec (
- std::map<DecisionNode *, FullVector> & p,
- DecisionNode *node );
- void calculateNodeProbabilities ( std::map<DecisionNode *, FullVector> & p, DecisionTree & tree );
- void calculateNodeProbVec
- ( std::map<DecisionNode *, FullVector> & nodeProbs,
- int classno,
-
-
- NICE::Vector & p );
- double calcInnerNodeProbs (
- DecisionNode *node,
- std::map<DecisionNode *, double> & p );
-
- void calcPosteriori (
- DecisionNode *node,
- const FullVector & apriori,
- const std::map<DecisionNode *, FullVector> & nodeprob,
- std::map<DecisionNode *, FullVector> & posterioriResult );
- void mapEstimateClass ( DecisionTree & tree,
- Examples & new_examples,
- int newClassNo,
- std::set<int> muClasses,
- std::set<int> substituteClasses,
- double sigmaq,
- int maxClassNo );
- public:
-
-
-
-
- DTEstimateAPriori( const NICE::Config *conf, const std::string & section );
-
-
- virtual ~DTEstimateAPriori();
- void reestimate ( DecisionTree & dt,
- Examples & examples,
- double sigmaq,
- int newClassNo,
- std::set<int> muClasses,
- std::set<int> substituteClasses,
- int maxClassNo );
-
- };
- }
- #endif
|