/data/refman/pasoursint/CMSSW_5_2_7_hltpatch2/src/CondFormats/EgammaObjects/interface/GBRForest.h

Go to the documentation of this file.

#ifndef EGAMMAOBJECTS_GBRForest
#define EGAMMAOBJECTS_GBRForest

//                                                                      //
// GBRForest                                                            //
//                                                                      //
// A fast minimal implementation of Gradient-Boosted Regression Trees   //
// which has been especially optimized for size on disk and in memory.  //                                                                  
//                                                                      //
// Designed to be built from TMVA-trained trees, but could also be      //
// generalized to otherwise-trained trees, classification,              //
//  or other boosting methods in the future                             //
//                                                                      //
//  Josh Bendavid - MIT                                                 //

#include <vector>
#include "GBRTree.h"


  namespace TMVA {
    class MethodBDT;
  }

  class GBRForest {

    public:

       GBRForest();
       explicit GBRForest(const TMVA::MethodBDT *bdt);
       virtual ~GBRForest();
       
       double GetResponse(const float* vector) const;
      
       std::vector<GBRTree> &Trees() { return fTrees; }
       
    protected:
      double               fInitialResponse;
      std::vector<GBRTree> fTrees;  
      
  };

//_______________________________________________________________________
inline double GBRForest::GetResponse(const float* vector) const {
  double response = fInitialResponse;
  for (std::vector<GBRTree>::const_iterator it=fTrees.begin(); it!=fTrees.end(); ++it) {
    response += it->GetResponse(vector);
  }
  return response;
}
  
#endif