BDT.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MVAUtils_BDT_H
#define MVAUtils_BDT_H
 
#include "MVAUtils/MVAUtilsDefs.h"
#include "TString.h"
#include <vector>
#include <map>
#include <cassert>
#include <memory>
#include "ForestBase.h"
 
class TTree;
namespace MVAUtils
{
 
  class BDT
  {
  public:
    explicit BDT(TTree *tree);//ctor TTree
   
    explicit BDT(std::unique_ptr<IForest> forest): 
      m_forest(std::move(forest)){
      }
 
    /* delete default ctor 
     * and default copy / assignment*/
    BDT() = delete;
    BDT (const BDT&) = delete;
    BDT& operator=(const BDT&) = delete;
    BDT (BDT&&) = default;
    BDT& operator=(BDT&&) = default;
    ~BDT()=default; 
 
 
 
    unsigned int GetNTrees() const ;
    int GetNVars() const ;
    float GetOffset() const ;
    float GetResponse(const std::vector<float>& values) const;
    float GetResponse(const std::vector<float*>& pointers) const;
    float GetResponse() const;
 
    float GetClassification(const std::vector<float>& values) const;
    float GetClassification(const std::vector<float*>& pointers) const;
    float GetClassification() const;
 
    // TMVA specific: return 2.0/(1.0+exp(-2.0*sum))-1, with no offset.
    float GetGradBoostMVA(const std::vector<float>& values) const;
    float GetGradBoostMVA(const std::vector<float*>& pointers) const;
 
    std::vector<float> GetMultiResponse(const std::vector<float>& values, unsigned int numClasses) const;
    std::vector<float> GetMultiResponse(const std::vector<float*>& pointers, unsigned int numClasses) const;
    std::vector<float> GetMultiResponse(unsigned int numClasses) const;
 
    std::vector<float> GetValues() const;
    const std::vector<float*>& GetPointers() const;
    void SetPointers(const std::vector<float*>& pointers);
 
    TTree* WriteTree(TString name = "BDT") const;
 
    void PrintForest() const;
    void PrintTree(unsigned int itree) const;
 
    float GetTreeResponse(const std::vector<float>& values, MVAUtils::index_t index) const;
    float GetTreeResponse(const std::vector<float*>& pointers, MVAUtils::index_t index) const;
 
  private:
    std::unique_ptr<IForest> m_forest;  
    std::vector<float*> m_pointers; 
  };
}
 
#include "MVAUtils/BDT.icc"
#endif