class MVAUtils::BDT #

[No description available] More…

#include <BDT.h>

Public Functions #

Detailed Description #

class MVAUtils::BDT;

Simplified Boosted Regression Tree, support TMVA, lgbm, and xgboost. Holds a forest (vector of top nodes of each decision tree) and a constant offset or set of weights (not always used).

Can be constructed from TMVA::MethodBDT or a TTree. Each entry of the TTree represents a binary tree and each element of the vectors stored in the TTree represent a node.

The response can be evaluated from a vector of floats or a vector of pointers (to avoid creating vectors at each call) which can be stored internally (m_pointers)

Public Functions Documentation #

function BDT #

explicit BDT(
    TTree * tree
)

Constructor. The input tree must be created with convertLGBMToRootTree.py (for lgbm training) or with convertXGBoostToRootTree.py (for xgboost training) or with convertXmlToRootTree (for tmva training)

function BDT #

explicit BDT(
    TMVA::MethodBDT * bdt,
    bool isRegression =true,
    bool useYesNoLeaf =false
)

c-tor from TMVA::MethodBDT

function ~BDT #

~BDT()

function GetNTrees #

unsigned int GetNTrees() const

Number of trees in the whole forest

function GetNVars #

int GetNVars() const

Number of variables expected in the inputs

function GetOffset #

float GetOffset() const

Get the offset to the whole forest

function GetResponse #

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

Get response of the forest, for regression

Return offset + the sum of the response of each tree

function GetResponse #

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

Return offset + the sum of the response of each tree

function GetResponse #

inline float GetResponse() const

function GetClassification #

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

Get response of the forest, for classification

function GetClassification #

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

function GetClassification #

inline float GetClassification() const

function GetGradBoostMVA #

float GetGradBoostMVA(
    const std::vector< float > & values
) const

function GetGradBoostMVA #

float GetGradBoostMVA(
    const std::vector< float * > & pointers
) const

function GetMultiResponse #

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

Get response of the forest, for multiclassification (e.g. b-tagging)

function GetMultiResponse #

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

function GetMultiResponse #

inline std::vector< float > GetMultiResponse(
    unsigned int numClasses
) const

function GetValues #

inline std::vector< float > GetValues() const

Return the values corresponding to m_pointers (or an empty vector)

function GetPointers #

inline const std::vector< float * > & GetPointers() const

Return stored pointers (which are used by methods with no args)

function SetPointers #

inline void SetPointers(
    const std::vector< float * > & pointers
)

Set the stored pointers so that one can use methods with no args

function WriteTree #

TTree * WriteTree(
    TString name ="BDT"
) const

Return a TTree representing the BDT: each entry is a binary tree, each element of the vectors is a node

function PrintForest #

void PrintForest() const

for debugging, print out tree or forest to stdout

function PrintTree #

void PrintTree(
    unsigned int itree
) const

function GetTreeResponse #

float GetTreeResponse(
    const std::vector< float > & values,
    MVAUtils::index_t index
) const

for debugging, return the response of a sigle tree given the index of its top node

function GetTreeResponse #

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

Updated on 2026-11-03 at 14:50:42 +0000