MVAUtils::ForestTMVA Class Reference

#include <ForestTMVA.h>

Inheritance diagram for MVAUtils::ForestTMVA:

Collaboration diagram for MVAUtils::ForestTMVA:

Public Member Functions
	ForestTMVA (TTree *tree)
	ForestTMVA ()=default
	ForestTMVA (const ForestTMVA &)=default
ForestTMVA &	operator= (const ForestTMVA &)=default
	ForestTMVA (ForestTMVA &&)=default
ForestTMVA &	operator= (ForestTMVA &&)=default
	~ForestTMVA ()=default
virtual TTree *	WriteTree (TString name) const override
	Return a TTree representing the BDT. More...
virtual float	GetResponse (const std::vector< float > &values) const override
	Compute the prediction for regression. More...
virtual float	GetResponse (const std::vector< float * > &pointers) const override
virtual float	GetClassification (const std::vector< float > &values) const override
	Compute the prediction of a classification. More...
virtual float	GetClassification (const std::vector< float * > &pointers) const override
virtual void	PrintForest () const override
virtual int	GetNVars () const override
	Get the number of input variable to be passed with std::vector to Get* methods. More...
void	setNVars (const int max_var)
float	GetTreeResponseWeighted (const std::vector< float > &values, unsigned int itree) const
float	GetTreeResponseWeighted (const std::vector< float * > &pointers, unsigned int itree) const
float	GetWeightedResponse (const std::vector< float > &values) const
float	GetWeightedResponse (const std::vector< float * > &pointers) const
void	newTree (const std::vector< NodeTMVA > &nodes, float weight)
float	GetTreeWeight (unsigned int itree) const
float	GetSumWeights () const
virtual float	GetOffset () const override
	Return the offset of the forest. More...
virtual float	GetOffset () const =0
	Return the global offset. More...
virtual void	PrintTree (unsigned int itree) const override
virtual float	GetTreeResponse (const std::vector< float > &values, unsigned int itree) const override final
	Return the response of one tree Must pass the features in a std::vector<float> values and the index of the tree. More...
virtual float	GetTreeResponse (const std::vector< float * > &pointers, unsigned int itree) const override final
virtual float	GetRawResponse (const std::vector< float > &values) const override final
	Return the response of the whole Forest. More...
virtual float	GetRawResponse (const std::vector< float * > &pointers) const override final
virtual std::vector< float >	GetMultiResponse (const std::vector< float > &values, unsigned int numClasses) const override
	Compute the prediction for multiclassification (a score for each class). More...
virtual std::vector< float >	GetMultiResponse (const std::vector< float * > &pointers, unsigned int numClasses) const override
virtual unsigned int	GetNTrees () const override final
virtual unsigned int	GetNTrees () const =0
virtual void	PrintForest () const override
std::vector< NodeTMVA >	GetTree (unsigned int itree) const
	Return the vector of nodes for the tree itree. More...

Protected Member Functions
void	newTree (const std::vector< NodeTMVA > &nodes)
	append a new tree (defined by a vector of nodes serialized in preorder) to the forest More...
float	GetTreeResponseFromNode (const std::vector< float > &values, index_t index) const
	Get the response of a tree. More...
float	GetTreeResponseFromNode (const std::vector< float * > &pointers, index_t index) const

Private Attributes
int	m_max_var =0
std::vector< float >	m_weights
	boost weights More...
float	m_sumWeights
	the sumOfBoostWeights–no need to recompute each call More...
std::vector< index_t >	m_forest
	indices of the top-level nodes of each tree More...
std::vector< NodeTMVA >	m_nodes
	where the nodes of the forest are stored More...

Detailed Description

Definition at line 63 of file ForestTMVA.h.

Constructor & Destructor Documentation

ForestTMVA() [1/4]

ForestTMVA::ForestTMVA ( TTree *  tree )

explicit

Definition at line 12 of file ForestTMVA.cxx.

  : ForestWeighted<NodeTMVA>()
  , m_max_var(0)
{
    // variables read from the TTree
    std::vector<int> *vars = nullptr;
    std::vector<float> *values = nullptr;
    float offset;  // the offset is the weight
 
    std::vector<NodeTMVA> nodes;
 
    tree->SetBranchAddress("vars", &vars);
    tree->SetBranchAddress("values", &values);
    tree->SetBranchAddress("offset", &offset);
 
    int numEntries = tree->GetEntries();
    for (int entry = 0; entry < numEntries; ++entry) {
      // each entry in the TTree is a decision tree
      tree->GetEntry(entry);
      if (!vars) {
        throw std::runtime_error(
          "vars pointer is null in ForestTMVA constructor");
      }
      if (!values) {
        throw std::runtime_error(
          "values pointers is null in ForestTMVA constructor");
      }
      if (vars->size() != values->size()) {
        throw std::runtime_error(
          "inconsistent size for vars and values in ForestTMVA constructor");
      }
 
      nodes.clear();
 
      std::vector<MVAUtils::index_t> right = detail::computeRight(*vars);
 
      for (size_t i = 0; i < vars->size(); ++i) {
        nodes.emplace_back(vars->at(i), values->at(i), right[i]);
        if (vars->at(i) > m_max_var) {
          m_max_var = vars->at(i);
        }
      }
      newTree(nodes, offset);
    } // end loop on TTree, all decision tree loaded
    delete vars;
    delete values;
}

ForestTMVA() [2/4]

MVAUtils::ForestTMVA::ForestTMVA ( )

default

ForestTMVA() [3/4]

MVAUtils::ForestTMVA::ForestTMVA ( const ForestTMVA &  )

default

ForestTMVA() [4/4]

MVAUtils::ForestTMVA::ForestTMVA ( ForestTMVA &&  )

default

~ForestTMVA()

MVAUtils::ForestTMVA::~ForestTMVA ( )

default

Member Function Documentation

GetClassification() [1/2]

virtual float MVAUtils::ForestTMVA::GetClassification ( const std::vector< float * > &  pointers ) const

overridevirtual

Implements MVAUtils::IForest.

GetClassification() [2/2]

virtual float MVAUtils::ForestTMVA::GetClassification ( const std::vector< float > &  values ) const

overridevirtual

Compute the prediction of a classification.

Implements MVAUtils::IForest.

GetMultiResponse() [1/2]

virtual std::vector<float> MVAUtils::Forest< NodeTMVA >::GetMultiResponse ( const std::vector< float * > &  pointers, unsigned int  numClasses  ) const

overridevirtualinherited

Implements MVAUtils::IForest.

GetMultiResponse() [2/2]

virtual std::vector<float> MVAUtils::Forest< NodeTMVA >::GetMultiResponse ( const std::vector< float > &  values, unsigned int  numClasses  ) const

overridevirtualinherited

Compute the prediction for multiclassification (a score for each class).

In addition to the input values need to pass the number of classes

Implements MVAUtils::IForest.

GetNTrees() [1/2]

virtual unsigned int MVAUtils::IForest::GetNTrees ( ) const

pure virtualinherited

Implemented in MVAUtils::Forest< Node_t >.

GetNTrees() [2/2]

virtual unsigned int MVAUtils::Forest< NodeTMVA >::GetNTrees

inlinefinaloverridevirtualinherited

Definition at line 94 of file Forest.h.

      {
        return m_forest.size();
      }

GetNVars()

virtual int MVAUtils::ForestTMVA::GetNVars ( ) const

inlineoverridevirtual

Get the number of input variable to be passed with std::vector to Get* methods.

Implements MVAUtils::IForest.

Definition at line 81 of file ForestTMVA.h.

{ return m_max_var + 1; }

GetOffset() [1/2]

virtual float MVAUtils::IForest::GetOffset ( ) const

pure virtualinherited

Return the global offset.

For many implementation this is just 0

Implemented in MVAUtils::Forest< Node_t >, and MVAUtils::ForestWeighted< Node_t >.

GetOffset() [2/2]

virtual float MVAUtils::ForestWeighted< NodeTMVA >::GetOffset

inlineoverridevirtualinherited

Return the offset of the forest.

Since by default there is no offset, return 0

Reimplemented from MVAUtils::Forest< NodeTMVA >.

Definition at line 41 of file ForestTMVA.h.

{ return m_weights[0]; }

GetRawResponse() [1/2]

virtual float MVAUtils::Forest< NodeTMVA >::GetRawResponse ( const std::vector< float * > &  pointers ) const

finaloverridevirtualinherited

Implements MVAUtils::IForest.

GetRawResponse() [2/2]

virtual float MVAUtils::Forest< NodeTMVA >::GetRawResponse ( const std::vector< float > &  values ) const

finaloverridevirtualinherited

Return the response of the whole Forest.

Raw is just the sum of all the trees

Implements MVAUtils::IForest.

GetResponse() [1/2]

virtual float MVAUtils::ForestTMVA::GetResponse ( const std::vector< float * > &  pointers ) const

overridevirtual

Reimplemented from MVAUtils::Forest< NodeTMVA >.

GetResponse() [2/2]

virtual float MVAUtils::ForestTMVA::GetResponse ( const std::vector< float > &  values ) const

overridevirtual

Compute the prediction for regression.

Reimplemented from MVAUtils::Forest< NodeTMVA >.

GetSumWeights()

float MVAUtils::ForestWeighted< NodeTMVA >::GetSumWeights

inlineinherited

Definition at line 39 of file ForestTMVA.h.

{ return m_sumWeights; }

GetTree()

std::vector<NodeTMVA > MVAUtils::Forest< NodeTMVA >::GetTree ( unsigned int  itree ) const

inherited

Return the vector of nodes for the tree itree.

GetTreeResponse() [1/2]

virtual float MVAUtils::Forest< NodeTMVA >::GetTreeResponse ( const std::vector< float * > &  pointers, unsigned int  itree  ) const

finaloverridevirtualinherited

Implements MVAUtils::IForest.

GetTreeResponse() [2/2]

virtual float MVAUtils::Forest< NodeTMVA >::GetTreeResponse ( const std::vector< float > &  values, unsigned int  itree  ) const

finaloverridevirtualinherited

Return the response of one tree Must pass the features in a std::vector<float> values and the index of the tree.

Implements MVAUtils::IForest.

GetTreeResponseFromNode() [1/2]

float MVAUtils::Forest< NodeTMVA >::GetTreeResponseFromNode ( const std::vector< float * > &  pointers, index_t  index  ) const

protectedinherited

GetTreeResponseFromNode() [2/2]

float MVAUtils::Forest< NodeTMVA >::GetTreeResponseFromNode ( const std::vector< float > &  values, index_t  index  ) const

protectedinherited

Get the response of a tree.

Instead of specifying the index of the tree (as in GetTreeResponse) the index of the top node of the tree should be specified

GetTreeResponseWeighted() [1/2]

float MVAUtils::ForestWeighted< NodeTMVA >::GetTreeResponseWeighted ( const std::vector< float * > &  pointers, unsigned int  itree  ) const

inherited

GetTreeResponseWeighted() [2/2]

float MVAUtils::ForestWeighted< NodeTMVA >::GetTreeResponseWeighted ( const std::vector< float > &  values, unsigned int  itree  ) const

inherited

GetTreeWeight()

float MVAUtils::ForestWeighted< NodeTMVA >::GetTreeWeight ( unsigned int  itree ) const

inlineinherited

Definition at line 38 of file ForestTMVA.h.

{ return m_weights[itree]; }

GetWeightedResponse() [1/2]

float MVAUtils::ForestWeighted< NodeTMVA >::GetWeightedResponse ( const std::vector< float * > &  pointers ) const

inherited

GetWeightedResponse() [2/2]

float MVAUtils::ForestWeighted< NodeTMVA >::GetWeightedResponse ( const std::vector< float > &  values ) const

inherited

newTree() [1/2]

void MVAUtils::Forest< NodeTMVA >::newTree ( const std::vector< NodeTMVA > &  nodes )

protectedinherited

append a new tree (defined by a vector of nodes serialized in preorder) to the forest

newTree() [2/2]

void MVAUtils::ForestWeighted< NodeTMVA >::newTree ( const std::vector< NodeTMVA > &  nodes, float  weight  )

inherited

operator=() [1/2]

ForestTMVA& MVAUtils::ForestTMVA::operator= ( const ForestTMVA &  )

default

operator=() [2/2]

ForestTMVA& MVAUtils::ForestTMVA::operator= ( ForestTMVA &&  )

default

PrintForest() [1/2]

void ForestTMVA::PrintForest ( ) const

overridevirtual

Implements MVAUtils::IForest.

Definition at line 86 of file ForestTMVA.cxx.

{
    std::cout << "***BDT TMVA: Printing entire forest***" << std::endl;
    Forest::PrintForest();
}

PrintForest() [2/2]

virtual void MVAUtils::Forest< NodeTMVA >::PrintForest

overridevirtualinherited

PrintTree()

virtual void MVAUtils::ForestWeighted< NodeTMVA >::PrintTree ( unsigned int  itree ) const

inlineoverridevirtualinherited

Reimplemented from MVAUtils::Forest< NodeTMVA >.

Definition at line 43 of file ForestTMVA.h.

                                                                  {
          std::cout << "weight: " << m_weights[itree] << std::endl;
          Forest<Node_t>::PrintTree(itree);
        }

setNVars()

void MVAUtils::ForestTMVA::setNVars ( const int  max_var )

inline

Definition at line 82 of file ForestTMVA.h.

{m_max_var=max_var;}

WriteTree()

TTree * ForestTMVA::WriteTree ( TString  ) const

overridevirtual

Return a TTree representing the BDT.

The called is the owner of the returned TTree

Implements MVAUtils::IForest.

Definition at line 61 of file ForestTMVA.cxx.

{
    TTree *tree = new TTree(name.Data(), "creator=TMVA");
 
    std::vector<int> vars;
    std::vector<float> values;
    float offset;
 
    tree->Branch("offset", &offset);
    tree->Branch("vars", &vars);
    tree->Branch("values", &values);
 
    for (size_t itree = 0; itree < GetNTrees(); ++itree) {
        vars.clear();
        values.clear();
        for(const auto& node : GetTree(itree)) {
            vars.push_back(node.GetVar());
            values.push_back(node.GetVal());
        }
        offset = GetTreeWeight(itree);
        tree->Fill();
    }
    return tree;
}

Member Data Documentation

m_forest

std::vector<index_t> MVAUtils::Forest< NodeTMVA >::m_forest

privateinherited

indices of the top-level nodes of each tree

Definition at line 117 of file Forest.h.

m_max_var

int MVAUtils::ForestTMVA::m_max_var =0

private

Definition at line 84 of file ForestTMVA.h.

m_nodes

std::vector<NodeTMVA > MVAUtils::Forest< NodeTMVA >::m_nodes

privateinherited

where the nodes of the forest are stored

Definition at line 118 of file Forest.h.

m_sumWeights

float MVAUtils::ForestWeighted< NodeTMVA >::m_sumWeights

privateinherited

the sumOfBoostWeights–no need to recompute each call

Definition at line 50 of file ForestTMVA.h.

m_weights

std::vector<float> MVAUtils::ForestWeighted< NodeTMVA >::m_weights

privateinherited

boost weights

Definition at line 49 of file ForestTMVA.h.

---

The documentation for this class was generated from the following files:

• ForestTMVA.h
• ForestTMVA.cxx