CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
List of all members | Public Member Functions | Private Attributes
TMVAEvaluator Class Reference

#include <TMVAEvaluator.h>

Public Member Functions

float evaluate (const std::map< std::string, float > &inputs, bool useSpectators=false) const
 
float evaluateGBRForest (const std::map< std::string, float > &inputs) const
 
float evaluateTMVA (const std::map< std::string, float > &inputs, bool useSpectators) const
 
void initialize (const std::string &options, const std::string &method, const std::string &weightFile, const std::vector< std::string > &variables, const std::vector< std::string > &spectators, bool useGBRForest=false, bool useAdaBoost=false)
 
void initializeGBRForest (const GBRForest *gbrForest, const std::vector< std::string > &variables, const std::vector< std::string > &spectators, bool useAdaBoost=false)
 
void initializeGBRForest (const edm::EventSetup &iSetup, const std::string &label, const std::vector< std::string > &variables, const std::vector< std::string > &spectators, bool useAdaBoost=false)
 
 TMVAEvaluator ()
 

Private Attributes

std::mutex m_mutex
 
std::shared_ptr< const GBRForestmGBRForest
 
std::unique_ptr< TMVA::IMethod > mIMethod
 
bool mIsInitialized
 
std::string mMethod
 
std::unique_ptr< TMVA::Reader > mReader
 
std::map< std::string,
std::pair< size_t, float > > 
mSpectators
 
bool mUseAdaBoost
 
bool mUsingGBRForest
 
std::map< std::string,
std::pair< size_t, float > > 
mVariables
 

Detailed Description

Definition at line 16 of file TMVAEvaluator.h.

Constructor & Destructor Documentation

TMVAEvaluator::TMVAEvaluator ( )

Definition at line 10 of file TMVAEvaluator.cc.

Member Function Documentation

float TMVAEvaluator::evaluate ( const std::map< std::string, float > &  inputs,
bool  useSpectators = false 
) const

Definition at line 150 of file TMVAEvaluator.cc.

float TMVAEvaluator::evaluateGBRForest ( const std::map< std::string, float > &  inputs) const

Definition at line 125 of file TMVAEvaluator.cc.

float TMVAEvaluator::evaluateTMVA ( const std::map< std::string, float > &  inputs,
bool  useSpectators 
) const

Definition at line 88 of file TMVAEvaluator.cc.

void TMVAEvaluator::initialize ( const std::string &  options,
const std::string &  method,
const std::string &  weightFile,
const std::vector< std::string > &  variables,
const std::vector< std::string > &  spectators,
bool  useGBRForest = false,
bool  useAdaBoost = false 
)

Definition at line 16 of file TMVAEvaluator.cc.

void TMVAEvaluator::initializeGBRForest ( const GBRForest gbrForest,
const std::vector< std::string > &  variables,
const std::vector< std::string > &  spectators,
bool  useAdaBoost = false 
)

Definition at line 57 of file TMVAEvaluator.cc.

void TMVAEvaluator::initializeGBRForest ( const edm::EventSetup iSetup,
const std::string &  label,
const std::vector< std::string > &  variables,
const std::vector< std::string > &  spectators,
bool  useAdaBoost = false 
)

Definition at line 77 of file TMVAEvaluator.cc.

Member Data Documentation

std::mutex TMVAEvaluator::m_mutex
mutableprivate

Definition at line 37 of file TMVAEvaluator.h.

std::shared_ptr<const GBRForest> TMVAEvaluator::mGBRForest
private

Definition at line 40 of file TMVAEvaluator.h.

std::unique_ptr<TMVA::IMethod> TMVAEvaluator::mIMethod
private

Definition at line 39 of file TMVAEvaluator.h.

bool TMVAEvaluator::mIsInitialized
private

Definition at line 32 of file TMVAEvaluator.h.

std::string TMVAEvaluator::mMethod
private

Definition at line 36 of file TMVAEvaluator.h.

std::unique_ptr<TMVA::Reader> TMVAEvaluator::mReader
private

Definition at line 38 of file TMVAEvaluator.h.

std::map<std::string,std::pair<size_t,float> > TMVAEvaluator::mSpectators
mutableprivate

Definition at line 43 of file TMVAEvaluator.h.

bool TMVAEvaluator::mUseAdaBoost
private

Definition at line 34 of file TMVAEvaluator.h.

bool TMVAEvaluator::mUsingGBRForest
private

Definition at line 33 of file TMVAEvaluator.h.

std::map<std::string,std::pair<size_t,float> > TMVAEvaluator::mVariables
mutableprivate

Definition at line 42 of file TMVAEvaluator.h.