19 const std::vector<std::string> &
variables,
const std::vector<std::string> & spectators)
22 mReader.reset(
new TMVA::Reader(options.c_str()));
27 for(std::vector<std::string>::const_iterator it = variables.begin(); it!=variables.end(); ++it)
29 mVariables.insert( std::pair<std::string,float>(*it,0.) );
34 for(std::vector<std::string>::const_iterator it = spectators.begin(); it!=spectators.end(); ++it)
36 mSpectators.insert( std::pair<std::string,float>(*it,0.) );
51 edm::LogError(
"InitializationError") <<
"TMVAEvaluator not properly initialized.";
57 edm::LogError(
"MissingInputVariable(s)") <<
"Too few input variables provided (" << inputs.size() <<
" provided but " <<
mVariables.size() <<
" expected).";
64 if (inputs.count(it->first)>0)
65 it->second = inputs.at(it->first);
67 edm::LogError(
"MissingInputVariable") <<
"Variable " << it->first <<
" is missing from the list of input variables. The returned discriminator value might not be sensible.";
std::map< std::string, float > mVariables
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)
std::unique_ptr< TMVA::Reader > mReader
float evaluate(const std::map< std::string, float > &inputs)
volatile std::atomic< bool > shutdown_flag false
void loadTMVAWeights(TMVA::Reader *reader, const std::string &method, const std::string &weightFile, bool verbose=false)
std::map< std::string, float > mSpectators