CMS 3D CMS Logo

List of all members | Public Types | Public Member Functions | Private Attributes
PileupJetIdAlgo::AlgoGBRForestsAndConstants Class Reference

#include <PileupJetIdAlgo.h>

Public Types

typedef float array_t[3][4][4]
 

Public Member Functions

 AlgoGBRForestsAndConstants (edm::ParameterSet const &, bool runMvas)
 
array_t const & betaStarCut () const
 
bool cutBased () const
 
bool etaBinnedWeights () const
 
std::vector< std::unique_ptr< const GBRForest > > const & etaReader () const
 
std::vector< double > const & jEtaMax () const
 
std::vector< double > const & jEtaMin () const
 
std::string const & label () const
 
array_t const & mvacut () const
 
int nEtaBins () const
 
std::unique_ptr< const GBRForest > const & reader () const
 
array_t const & rmsCut () const
 
bool runMvas () const
 
std::vector< std::vector< std::string > > const & tmvaEtaVariables () const
 
std::string const & tmvaMethod () const
 
std::vector< std::string > const & tmvaVariables () const
 

Private Attributes

float betaStarCut_ [3][4][4]
 
bool cutBased_
 
bool etaBinnedWeights_
 
std::vector< std::unique_ptr< const GBRForest > > etaReader_
 
std::vector< double > jEtaMax_
 
std::vector< double > jEtaMin_
 
std::string label_
 
float mvacut_ [3][4][4]
 
int nEtaBins_
 
std::unique_ptr< const GBRForestreader_
 
float rmsCut_ [3][4][4]
 
bool runMvas_
 
std::vector< std::vector< std::string > > tmvaEtaVariables_
 
std::string tmvaMethod_
 
std::map< std::string, std::string > tmvaNames_
 
std::vector< std::string > tmvaVariables_
 

Detailed Description

Definition at line 56 of file PileupJetIdAlgo.h.

Member Typedef Documentation

typedef float PileupJetIdAlgo::AlgoGBRForestsAndConstants::array_t[3][4][4]

Definition at line 74 of file PileupJetIdAlgo.h.

Constructor & Destructor Documentation

PileupJetIdAlgo::AlgoGBRForestsAndConstants::AlgoGBRForestsAndConstants ( edm::ParameterSet const &  ps,
bool  runMvas 
)

Definition at line 21 of file PileupJetIdAlgo.cc.

References betaStarCut_, createGBRForest(), cutBased_, etaBinnedWeights_, etaReader_, edm::ParameterSet::getParameter(), jEtaMax_, jEtaMin_, PileupJetIdentifier::kLoose, PileupJetIdentifier::kMedium, mvacut_, nEtaBins_, reader_, rmsCut_, runMvas_, AlCaHLTBitMon_QueryRunRegistry::string, tmvaEtaVariables_, tmvaMethod_, hltPUIdAlgo_cff::tmvaSpectators, tmvaVariables_, PileupJetIDParams_cfi::trainings, PileupJetIdAlgo::USER, findQualityFiles::v, and jets_cff::version.

21  :
22  cutBased_(ps.getParameter<bool>("cutBased")),
23  etaBinnedWeights_(false),
25  nEtaBins_(0),
26  label_(ps.getParameter<std::string>("label")),
27  mvacut_{},
28  rmsCut_{},
29  betaStarCut_{}
30  {
31 
32  std::vector<edm::FileInPath> tmvaEtaWeights;
33  std::vector<std::string> tmvaSpectators;
34  int version;
35 
36  if (!cutBased_) {
37  etaBinnedWeights_ = ps.getParameter<bool>("etaBinnedWeights");
38  if (etaBinnedWeights_) {
39  const std::vector<edm::ParameterSet>& trainings = ps.getParameter<std::vector <edm::ParameterSet> >("trainings");
40  nEtaBins_ = ps.getParameter<int>("nEtaBins");
41  for (int v = 0; v < nEtaBins_; v++) {
42  tmvaEtaWeights.push_back(trainings.at(v).getParameter<edm::FileInPath>("tmvaWeights"));
43  jEtaMin_.push_back( trainings.at(v).getParameter<double>("jEtaMin") );
44  jEtaMax_.push_back( trainings.at(v).getParameter<double>("jEtaMax") );
45  }
46  for (int v = 0; v < nEtaBins_; v++) {
47  tmvaEtaVariables_.push_back( trainings.at(v).getParameter<std::vector<std::string> >("tmvaVariables") );
48  }
49  } else {
50  tmvaVariables_ = ps.getParameter<std::vector<std::string> >("tmvaVariables");
51  }
52  tmvaMethod_ = ps.getParameter<std::string>("tmvaMethod");
53  tmvaSpectators = ps.getParameter<std::vector<std::string> >("tmvaSpectators");
54  version = ps.getParameter<int>("version");
55  } else {
56  version = USER;
57  }
58 
59  edm::ParameterSet jetConfig = ps.getParameter<edm::ParameterSet>("JetIdParams");
60  for (int i0 = 0; i0 < 3; i0++) {
61  std::string lCutType = "Tight";
62  if (i0 == PileupJetIdentifier::kMedium) lCutType = "Medium";
63  if (i0 == PileupJetIdentifier::kLoose) lCutType = "Loose";
64  int nCut = 1;
65  if(cutBased_) nCut++;
66  for (int i1 = 0; i1 < nCut; i1++) {
67  std::string lFullCutType = lCutType;
68  if (cutBased_ && i1 == 0) lFullCutType = "BetaStar"+ lCutType;
69  if (cutBased_ && i1 == 1) lFullCutType = "RMS" + lCutType;
70  std::vector<double> pt010 = jetConfig.getParameter<std::vector<double> >(("Pt010_" +lFullCutType).c_str());
71  std::vector<double> pt1020 = jetConfig.getParameter<std::vector<double> >(("Pt1020_"+lFullCutType).c_str());
72  std::vector<double> pt2030 = jetConfig.getParameter<std::vector<double> >(("Pt2030_"+lFullCutType).c_str());
73  std::vector<double> pt3050 = jetConfig.getParameter<std::vector<double> >(("Pt3050_"+lFullCutType).c_str());
74  if (!cutBased_) {
75  for (int i2 = 0; i2 < 4; i2++) mvacut_[i0][0][i2] = pt010 [i2];
76  for (int i2 = 0; i2 < 4; i2++) mvacut_[i0][1][i2] = pt1020[i2];
77  for (int i2 = 0; i2 < 4; i2++) mvacut_[i0][2][i2] = pt2030[i2];
78  for (int i2 = 0; i2 < 4; i2++) mvacut_[i0][3][i2] = pt3050[i2];
79  }
80  if (cutBased_ && i1 == 0) {
81  for (int i2 = 0; i2 < 4; i2++) betaStarCut_[i0][0][i2] = pt010 [i2];
82  for (int i2 = 0; i2 < 4; i2++) betaStarCut_[i0][1][i2] = pt1020[i2];
83  for (int i2 = 0; i2 < 4; i2++) betaStarCut_[i0][2][i2] = pt2030[i2];
84  for (int i2 = 0; i2 < 4; i2++) betaStarCut_[i0][3][i2] = pt3050[i2];
85  }
86  if (cutBased_ && i1 == 1) {
87  for (int i2 = 0; i2 < 4; i2++) rmsCut_[i0][0][i2] = pt010 [i2];
88  for (int i2 = 0; i2 < 4; i2++) rmsCut_[i0][1][i2] = pt1020[i2];
89  for (int i2 = 0; i2 < 4; i2++) rmsCut_[i0][2][i2] = pt2030[i2];
90  for (int i2 = 0; i2 < 4; i2++) rmsCut_[i0][3][i2] = pt3050[i2];
91  }
92  }
93  }
94 
95  if ( ! cutBased_ ) {
96  assert( tmvaMethod_.empty() || ((! tmvaVariables_.empty() || ( !tmvaEtaVariables_.empty() )) && version == USER) );
97  }
98 
99  if (( ! cutBased_ ) && (runMvas_)) {
100  if (etaBinnedWeights_) {
101  for (int v = 0; v < nEtaBins_; v++) {
102  etaReader_.push_back(createGBRForest(tmvaEtaWeights.at(v)));
103  }
104  } else {
105  reader_ = createGBRForest(ps.getParameter<std::string>("tmvaWeights"));
106  }
107  }
108 }
T getParameter(std::string const &) const
std::vector< std::unique_ptr< const GBRForest > > etaReader_
std::vector< std::string > tmvaVariables_
std::vector< std::vector< std::string > > tmvaEtaVariables_
std::unique_ptr< const GBRForest > reader_
std::unique_ptr< const GBRForest > createGBRForest(const std::string &weightsFile)

Member Function Documentation

array_t const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::betaStarCut ( ) const
inline

Definition at line 77 of file PileupJetIdAlgo.h.

References betaStarCut_.

bool PileupJetIdAlgo::AlgoGBRForestsAndConstants::cutBased ( ) const
inline

Definition at line 63 of file PileupJetIdAlgo.h.

References cutBased_.

Referenced by PileupJetIdAlgo::runMva().

bool PileupJetIdAlgo::AlgoGBRForestsAndConstants::etaBinnedWeights ( ) const
inline

Definition at line 64 of file PileupJetIdAlgo.h.

References etaBinnedWeights_.

Referenced by PileupJetIdAlgo::runMva().

std::vector<std::unique_ptr<const GBRForest> > const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::etaReader ( ) const
inline

Definition at line 62 of file PileupJetIdAlgo.h.

References etaReader_.

Referenced by PileupJetIdAlgo::runMva().

62 { return etaReader_; }
std::vector< std::unique_ptr< const GBRForest > > etaReader_
std::vector<double> const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::jEtaMax ( ) const
inline

Definition at line 68 of file PileupJetIdAlgo.h.

References jEtaMax_.

Referenced by PileupJetIdAlgo::runMva().

68 { return jEtaMax_; }
std::vector<double> const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::jEtaMin ( ) const
inline

Definition at line 67 of file PileupJetIdAlgo.h.

References jEtaMin_.

Referenced by PileupJetIdAlgo::runMva().

67 { return jEtaMin_; }
std::string const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::label ( ) const
inline

Definition at line 69 of file PileupJetIdAlgo.h.

References label_.

array_t const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::mvacut ( ) const
inline

Definition at line 75 of file PileupJetIdAlgo.h.

References mvacut_.

Referenced by PileupJetIdAlgo::computeIDflag().

int PileupJetIdAlgo::AlgoGBRForestsAndConstants::nEtaBins ( ) const
inline

Definition at line 66 of file PileupJetIdAlgo.h.

References nEtaBins_.

Referenced by PileupJetIdAlgo::runMva().

std::unique_ptr<const GBRForest> const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::reader ( ) const
inline

Definition at line 61 of file PileupJetIdAlgo.h.

References reader_.

Referenced by PileupJetIdAlgo::runMva().

61 { return reader_; }
std::unique_ptr< const GBRForest > reader_
array_t const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::rmsCut ( ) const
inline

Definition at line 76 of file PileupJetIdAlgo.h.

References rmsCut_.

bool PileupJetIdAlgo::AlgoGBRForestsAndConstants::runMvas ( ) const
inline

Definition at line 65 of file PileupJetIdAlgo.h.

References runMvas_.

Referenced by PileupJetIdAlgo::computeIdVariables().

std::vector<std::vector<std::string> > const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::tmvaEtaVariables ( ) const
inline

Definition at line 72 of file PileupJetIdAlgo.h.

References tmvaEtaVariables_.

Referenced by PileupJetIdAlgo::runMva().

72 { return tmvaEtaVariables_; }
std::vector< std::vector< std::string > > tmvaEtaVariables_
std::string const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::tmvaMethod ( ) const
inline

Definition at line 70 of file PileupJetIdAlgo.h.

References tmvaMethod_.

Referenced by PileupJetIdAlgo::method().

std::vector<std::string> const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::tmvaVariables ( ) const
inline

Definition at line 71 of file PileupJetIdAlgo.h.

References tmvaVariables_.

Referenced by PileupJetIdAlgo::runMva().

71 { return tmvaVariables_; }
std::vector< std::string > tmvaVariables_

Member Data Documentation

float PileupJetIdAlgo::AlgoGBRForestsAndConstants::betaStarCut_[3][4][4]
private

Definition at line 96 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and betaStarCut().

bool PileupJetIdAlgo::AlgoGBRForestsAndConstants::cutBased_
private

Definition at line 83 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and cutBased().

bool PileupJetIdAlgo::AlgoGBRForestsAndConstants::etaBinnedWeights_
private

Definition at line 84 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and etaBinnedWeights().

std::vector<std::unique_ptr<const GBRForest> > PileupJetIdAlgo::AlgoGBRForestsAndConstants::etaReader_
private

Definition at line 82 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and etaReader().

std::vector<double> PileupJetIdAlgo::AlgoGBRForestsAndConstants::jEtaMax_
private

Definition at line 88 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and jEtaMax().

std::vector<double> PileupJetIdAlgo::AlgoGBRForestsAndConstants::jEtaMin_
private

Definition at line 87 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and jEtaMin().

std::string PileupJetIdAlgo::AlgoGBRForestsAndConstants::label_
private
float PileupJetIdAlgo::AlgoGBRForestsAndConstants::mvacut_[3][4][4]
private

Definition at line 94 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and mvacut().

int PileupJetIdAlgo::AlgoGBRForestsAndConstants::nEtaBins_
private

Definition at line 86 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and nEtaBins().

std::unique_ptr<const GBRForest> PileupJetIdAlgo::AlgoGBRForestsAndConstants::reader_
private

Definition at line 81 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and reader().

float PileupJetIdAlgo::AlgoGBRForestsAndConstants::rmsCut_[3][4][4]
private

Definition at line 95 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and rmsCut().

bool PileupJetIdAlgo::AlgoGBRForestsAndConstants::runMvas_
private

Definition at line 85 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and runMvas().

std::vector<std::vector<std::string> > PileupJetIdAlgo::AlgoGBRForestsAndConstants::tmvaEtaVariables_
private

Definition at line 92 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and tmvaEtaVariables().

std::string PileupJetIdAlgo::AlgoGBRForestsAndConstants::tmvaMethod_
private

Definition at line 90 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and tmvaMethod().

std::map<std::string,std::string> PileupJetIdAlgo::AlgoGBRForestsAndConstants::tmvaNames_
private

Definition at line 98 of file PileupJetIdAlgo.h.

std::vector<std::string> PileupJetIdAlgo::AlgoGBRForestsAndConstants::tmvaVariables_
private

Definition at line 91 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and tmvaVariables().