LArSoft
v09_90_00
Liquid Argon Software toolkit - https://larsoft.org/
|
AdaBoostDecisionTree class. More...
#include "LArAdaBoostDecisionTree.h"
Classes | |
class | Node |
Node class used for representing a decision tree. More... | |
class | StrongClassifier |
StrongClassifier class used in application of adaptive boost decision tree. More... | |
class | WeakClassifier |
WeakClassifier class containing a decision tree and a weight. More... | |
Public Member Functions | |
AdaBoostDecisionTree () | |
Constructor. More... | |
AdaBoostDecisionTree (const AdaBoostDecisionTree &rhs) | |
Copy constructor. More... | |
AdaBoostDecisionTree & | operator= (const AdaBoostDecisionTree &rhs) |
Assignment operator. More... | |
~AdaBoostDecisionTree () | |
Destructor. More... | |
pandora::StatusCode | Initialize (const std::string ¶meterLocation, const std::string &bdtName) |
Initialize the bdt model. More... | |
bool | Classify (const LArMvaHelper::MvaFeatureVector &features) const |
Classify the set of input features based on the trained model. More... | |
double | CalculateClassificationScore (const LArMvaHelper::MvaFeatureVector &features) const |
Calculate the classification score for a set of input features, based on the trained model. More... | |
double | CalculateProbability (const LArMvaHelper::MvaFeatureVector &features) const |
Calculate the classification probability for a set of input features, based on the trained model. More... | |
Private Types | |
typedef std::map< int, const Node * > | IdToNodeMap |
typedef std::vector< const WeakClassifier * > | WeakClassifiers |
Private Member Functions | |
double | CalculateScore (const LArMvaHelper::MvaFeatureVector &features) const |
Calculate score for input features using strong classifier. More... | |
Private Attributes | |
StrongClassifier * | m_pStrongClassifier |
Strong adaptive boost tree classifier. More... | |
AdaBoostDecisionTree class.
Definition at line 27 of file LArAdaBoostDecisionTree.h.
|
private |
Definition at line 191 of file LArAdaBoostDecisionTree.h.
|
private |
Definition at line 264 of file LArAdaBoostDecisionTree.h.
lar_content::AdaBoostDecisionTree::AdaBoostDecisionTree | ( | ) |
Constructor.
Definition at line 18 of file LArAdaBoostDecisionTree.cc.
lar_content::AdaBoostDecisionTree::AdaBoostDecisionTree | ( | const AdaBoostDecisionTree & | rhs | ) |
Copy constructor.
rhs | the AdaBoostDecisionTree to copy |
Definition at line 25 of file LArAdaBoostDecisionTree.cc.
References m_pStrongClassifier.
lar_content::AdaBoostDecisionTree::~AdaBoostDecisionTree | ( | ) |
Destructor.
Definition at line 42 of file LArAdaBoostDecisionTree.cc.
References m_pStrongClassifier.
|
virtual |
Calculate the classification score for a set of input features, based on the trained model.
features | the input features |
Implements lar_content::MvaInterface.
Definition at line 127 of file LArAdaBoostDecisionTree.cc.
References CalculateScore().
|
virtual |
Calculate the classification probability for a set of input features, based on the trained model.
features | the input features |
Implements lar_content::MvaInterface.
Definition at line 134 of file LArAdaBoostDecisionTree.cc.
References CalculateScore().
|
private |
Calculate score for input features using strong classifier.
features | the input features |
Definition at line 143 of file LArAdaBoostDecisionTree.cc.
References m_pStrongClassifier, lar_content::AdaBoostDecisionTree::Node::Node(), and lar_content::AdaBoostDecisionTree::StrongClassifier::Predict().
Referenced by CalculateClassificationScore(), CalculateProbability(), and Classify().
|
virtual |
Classify the set of input features based on the trained model.
features | the input features |
Implements lar_content::MvaInterface.
Definition at line 120 of file LArAdaBoostDecisionTree.cc.
References CalculateScore().
StatusCode lar_content::AdaBoostDecisionTree::Initialize | ( | const std::string & | parameterLocation, |
const std::string & | bdtName | ||
) |
Initialize the bdt model.
parameterLocation | the location of the model |
bdtName | the name of the model |
Definition at line 49 of file LArAdaBoostDecisionTree.cc.
References m_pStrongClassifier.
Referenced by lar_content::BdtBeamParticleIdTool::ReadSettings().
AdaBoostDecisionTree & lar_content::AdaBoostDecisionTree::operator= | ( | const AdaBoostDecisionTree & | rhs | ) |
Assignment operator.
rhs | the AdaBoostDecisionTree to assign |
Definition at line 32 of file LArAdaBoostDecisionTree.cc.
References m_pStrongClassifier.
|
private |
Strong adaptive boost tree classifier.
Definition at line 325 of file LArAdaBoostDecisionTree.h.
Referenced by AdaBoostDecisionTree(), CalculateScore(), Initialize(), operator=(), and ~AdaBoostDecisionTree().