MVAUtils::ForestXGBoost Class Reference

Implement XGBoost with nan support. More...

#include <ForestXGBoost.h>

Inheritance diagram for MVAUtils::ForestXGBoost:

Collaboration diagram for MVAUtils::ForestXGBoost:

Public Member Functions
	ForestXGBoost (TTree *tree)
	ForestXGBoost ()=default
	ForestXGBoost (const ForestXGBoost &)=default
ForestXGBoost &	operator= (const ForestXGBoost &)=default
	ForestXGBoost (ForestXGBoost &&)=default
ForestXGBoost &	operator= (ForestXGBoost &&)=default
	~ForestXGBoost ()=default
virtual TTree *	WriteTree (TString name) const override
	Return a TTree representing the BDT.
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.
virtual float	GetClassification (const std::vector< float > &values) const final
	Compute the prediction of a classification.
virtual float	GetClassification (const std::vector< float * > &pointers) const=0
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.
virtual float	GetTreeResponse (const std::vector< float * > &pointers, unsigned int itree) const =0
virtual float	GetOffset () const override
	Return the offset of the forest.
virtual float	GetRawResponse (const std::vector< float > &values) const override final
	Return the response of the whole Forest.
virtual float	GetRawResponse (const std::vector< float * > &pointers) const =0
virtual float	GetResponse (const std::vector< float > &values) const override
	Compute the prediction for regression.
virtual float	GetResponse (const std::vector< float * > &pointers) const =0
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).
virtual std::vector< float >	GetMultiResponse (const std::vector< float * > &pointers, unsigned int numClasses) const =0
virtual unsigned int	GetNTrees () const override final
virtual void	PrintTree (unsigned int itree) const override
std::vector< NodeXGBoost >	GetTree (unsigned int itree) const
	Return the vector of nodes for the tree itree.

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

Private Attributes
int	m_max_var =0
std::vector< index_t >	m_forest
	indices of the top-level nodes of each tree
std::vector< NodeXGBoost >	m_nodes
	where the nodes of the forest are stored

Detailed Description

Implement XGBoost with nan support.

Definition at line 44 of file ForestXGBoost.h.

Constructor & Destructor Documentation

ForestXGBoost() [1/4]

ForestXGBoost::ForestXGBoost ( TTree * tree )

explicit

Definition at line 12 of file ForestXGBoost.cxx.

  : ForestXGBoostBase<NodeXGBoost>()
  , m_max_var(0)
{
    
 
    // variables read from the TTree
    std::vector<int> *vars = nullptr;
    std::vector<float> *values = nullptr;
    std::vector<bool> *default_left = nullptr;
 
    std::vector<NodeXGBoost> nodes;
 
    tree->SetBranchAddress("vars", &vars);
    tree->SetBranchAddress("values", &values);
    tree->SetBranchAddress("default_left", &default_left);
 
    for (int i = 0; i < tree->GetEntries(); ++i)
    {
        // each entry in the TTree is a decision tree
        tree->GetEntry(i);
        if (!vars) { throw std::runtime_error("vars pointer is null in ForestXGBoost constructor"); }
        if (!values) { throw std::runtime_error("values pointers is null in ForestXGBoost constructor"); }
        if (!default_left) { throw std::runtime_error("default_left pointers is null in ForestXGBoost constructor"); }
        if (vars->size() != values->size()) { throw std::runtime_error("inconsistent size for vars and values in ForestXGBoost constructor"); }
        if (default_left->size() != values->size()) { throw std::runtime_error("inconsistent size for default_left and values in ForestXGBoost 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], default_left->at(i));
            if (vars->at(i) > m_max_var) { m_max_var = vars->at(i); }
        }
        newTree(nodes);
    }  // end loop on TTree, all decision tree loaded
    delete vars;
    delete values;
    delete default_left;
}

ForestXGBoost() [2/4]

MVAUtils::ForestXGBoost::ForestXGBoost ( )

default

ForestXGBoost() [3/4]

MVAUtils::ForestXGBoost::ForestXGBoost ( const ForestXGBoost & )

default

ForestXGBoost() [4/4]

MVAUtils::ForestXGBoost::ForestXGBoost ( ForestXGBoost && )

default

~ForestXGBoost()

MVAUtils::ForestXGBoost::~ForestXGBoost ( )

default

Member Function Documentation

GetClassification() [1/2]

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

pure virtualinherited

GetClassification() [2/2]

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

inlinefinalvirtualinherited

Compute the prediction of a classification.

Implements MVAUtils::IForest.

Definition at line 33 of file ForestXGBoost.h.

        {
            return detail::sigmoid(GetResponse(values));
        }

GetMultiResponse() [1/2]

virtual std::vector< float > MVAUtils::Forest< NodeXGBoost >::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.

GetMultiResponse() [2/2]

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

pure virtualinherited

Implemented in MVAUtils::Forest< Node_t >.

GetNTrees()

virtual unsigned int MVAUtils::Forest< NodeXGBoost >::GetNTrees ( ) const

inlinefinaloverridevirtualinherited

Implements MVAUtils::IForest.

Definition at line 94 of file Forest.h.

      {
        return m_forest.size();
      }

GetNVars()

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

inlineoverridevirtual

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

Implements MVAUtils::IForest.

Definition at line 57 of file ForestXGBoost.h.

{ return m_max_var + 1; }

GetOffset()

virtual float MVAUtils::Forest< NodeXGBoost >::GetOffset ( ) const

inlineoverridevirtualinherited

Return the offset of the forest.

Since by default there is no offset, return 0

Implements MVAUtils::IForest.

Definition at line 63 of file Forest.h.

{ return 0.; }

GetRawResponse() [1/2]

virtual float MVAUtils::Forest< NodeXGBoost >::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.

GetRawResponse() [2/2]

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

pure virtualinherited

Implemented in MVAUtils::Forest< Node_t >.

GetResponse() [1/2]

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

overridevirtualinherited

Compute the prediction for regression.

Implements MVAUtils::IForest.

GetResponse() [2/2]

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

pure virtualinherited

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

GetTree()

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

inherited

Return the vector of nodes for the tree itree.

GetTreeResponse() [1/2]

virtual float MVAUtils::Forest< NodeXGBoost >::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.

GetTreeResponse() [2/2]

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

pure virtualinherited

Implemented in MVAUtils::Forest< Node_t >.

GetTreeResponseFromNode()

float MVAUtils::Forest< NodeXGBoost >::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

newTree()

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

protectedinherited

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

operator=() [1/2]

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

default

operator=() [2/2]

ForestXGBoost & MVAUtils::ForestXGBoost::operator= ( ForestXGBoost && )

default

PrintForest()

void ForestXGBoost::PrintForest ( ) const

overridevirtual

Reimplemented from MVAUtils::Forest< NodeXGBoost >.

Definition at line 81 of file ForestXGBoost.cxx.

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

PrintTree()

virtual void MVAUtils::Forest< NodeXGBoost >::PrintTree ( unsigned int itree ) const

overridevirtualinherited

Implements MVAUtils::IForest.

WriteTree()

TTree * ForestXGBoost::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 55 of file ForestXGBoost.cxx.

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

Member Data Documentation

m_forest

std::vector<index_t> MVAUtils::Forest< NodeXGBoost >::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::ForestXGBoost::m_max_var =0

private

Definition at line 59 of file ForestXGBoost.h.

m_nodes

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

privateinherited

where the nodes of the forest are stored

Definition at line 118 of file Forest.h.

---

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

• ForestXGBoost.h
• ForestXGBoost.cxx