|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
#ifndef CommonTools_MVAUtils_TMVAEvaluator_h
#define CommonTools_MVAUtils_TMVAEvaluator_h
#include <map>
#include <memory>
#include <mutex>
#include <string>
#include <vector>
#include "CondFormats/GBRForest/interface/GBRForest.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Utilities/interface/thread_safety_macros.h"
#include "TMVA/IMethod.h"
#include "TMVA/Reader.h"
class TMVAEvaluator {
public:
TMVAEvaluator();
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);
float evaluateTMVA(const std::map<std::string, float>& inputs, bool useSpectators) const;
float evaluateGBRForest(const std::map<std::string, float>& inputs) const;
float evaluate(const std::map<std::string, float>& inputs, bool useSpectators = false) const;
private:
bool mIsInitialized;
bool mUsingGBRForest;
bool mUseAdaBoost;
std::string mMethod;
mutable std::mutex m_mutex;
CMS_THREAD_GUARD(m_mutex) std::unique_ptr<TMVA::Reader> mReader;
std::shared_ptr<const GBRForest> mGBRForest;
CMS_THREAD_GUARD(m_mutex) mutable std::map<std::string, std::pair<size_t, float>> mVariables;
CMS_THREAD_GUARD(m_mutex) mutable std::map<std::string, std::pair<size_t, float>> mSpectators;
};
#endif // CommonTools_Utils_TMVAEvaluator_h
|