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::unique_ptr< const GBRForest >	getMVA (std::vector< std::string > const &varList, std::string const &tmvaWeights, std::vector< std::string > const &tmvaSpectators)
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 GBRForest >	reader_
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 61 of file PileupJetIdAlgo.h.

Member Typedef Documentation

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

Definition at line 79 of file PileupJetIdAlgo.h.

Constructor & Destructor Documentation

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

Definition at line 22 of file PileupJetIdAlgo.cc.

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

                                                                                                             :
  cutBased_(ps.getParameter<bool>("cutBased")),
  etaBinnedWeights_(false),
  runMvas_(runMvas),
  nEtaBins_(0),
  label_(ps.getParameter<std::string>("label")),
  mvacut_{},
  rmsCut_{},
  betaStarCut_{}
 {

  std::string tmvaWeights;
  std::vector<std::string> tmvaEtaWeights;
  std::vector<std::string> tmvaSpectators;
  int version;

  if (!cutBased_) {
    etaBinnedWeights_ = ps.getParameter<bool>("etaBinnedWeights");
    if (etaBinnedWeights_) {
      const std::vector<edm::ParameterSet>& trainings = ps.getParameter<std::vector <edm::ParameterSet> >("trainings");
      nEtaBins_ = ps.getParameter<int>("nEtaBins");
      for (int v = 0; v < nEtaBins_; v++) {
        tmvaEtaWeights.push_back( edm::FileInPath(trainings.at(v).getParameter<std::string>("tmvaWeights")).fullPath() );
        jEtaMin_.push_back( trainings.at(v).getParameter<double>("jEtaMin") );
        jEtaMax_.push_back( trainings.at(v).getParameter<double>("jEtaMax") );
      }
      for (int v = 0; v < nEtaBins_; v++) {
        tmvaEtaVariables_.push_back( trainings.at(v).getParameter<std::vector<std::string> >("tmvaVariables") );
      }
    } else {
      tmvaWeights = edm::FileInPath(ps.getParameter<std::string>("tmvaWeights")).fullPath();
      tmvaVariables_ = ps.getParameter<std::vector<std::string> >("tmvaVariables");
    }
    tmvaMethod_ = ps.getParameter<std::string>("tmvaMethod");
    tmvaSpectators = ps.getParameter<std::vector<std::string> >("tmvaSpectators");
    version = ps.getParameter<int>("version");
  } else {
    version = USER;
  }

  edm::ParameterSet jetConfig = ps.getParameter<edm::ParameterSet>("JetIdParams");
  for (int i0 = 0; i0 < 3; i0++) {
    std::string lCutType                             = "Tight";
    if (i0 == PileupJetIdentifier::kMedium) lCutType = "Medium";
    if (i0 == PileupJetIdentifier::kLoose)  lCutType = "Loose";
    int nCut = 1;
    if(cutBased_) nCut++;
    for (int i1 = 0; i1 < nCut; i1++) {
      std::string lFullCutType = lCutType;
      if (cutBased_ && i1 == 0) lFullCutType = "BetaStar"+ lCutType;
      if (cutBased_ && i1 == 1) lFullCutType = "RMS"     + lCutType;
      std::vector<double> pt010  = jetConfig.getParameter<std::vector<double> >(("Pt010_" +lFullCutType).c_str());
      std::vector<double> pt1020 = jetConfig.getParameter<std::vector<double> >(("Pt1020_"+lFullCutType).c_str());
      std::vector<double> pt2030 = jetConfig.getParameter<std::vector<double> >(("Pt2030_"+lFullCutType).c_str());
      std::vector<double> pt3050 = jetConfig.getParameter<std::vector<double> >(("Pt3050_"+lFullCutType).c_str());
      if (!cutBased_) {
        for (int i2 = 0; i2 < 4; i2++) mvacut_[i0][0][i2] = pt010 [i2];
        for (int i2 = 0; i2 < 4; i2++) mvacut_[i0][1][i2] = pt1020[i2];
        for (int i2 = 0; i2 < 4; i2++) mvacut_[i0][2][i2] = pt2030[i2];
        for (int i2 = 0; i2 < 4; i2++) mvacut_[i0][3][i2] = pt3050[i2];
      }
      if (cutBased_ && i1 == 0) {
        for (int i2 = 0; i2 < 4; i2++) betaStarCut_[i0][0][i2] = pt010 [i2];
        for (int i2 = 0; i2 < 4; i2++) betaStarCut_[i0][1][i2] = pt1020[i2];
        for (int i2 = 0; i2 < 4; i2++) betaStarCut_[i0][2][i2] = pt2030[i2];
        for (int i2 = 0; i2 < 4; i2++) betaStarCut_[i0][3][i2] = pt3050[i2];
      }
      if (cutBased_ && i1 == 1) {
        for (int i2 = 0; i2 < 4; i2++) rmsCut_[i0][0][i2] = pt010 [i2];
        for (int i2 = 0; i2 < 4; i2++) rmsCut_[i0][1][i2] = pt1020[i2];
        for (int i2 = 0; i2 < 4; i2++) rmsCut_[i0][2][i2] = pt2030[i2];
        for (int i2 = 0; i2 < 4; i2++) rmsCut_[i0][3][i2] = pt3050[i2];
      }
    }
  }

  if ( ! cutBased_ ) {
    assert( tmvaMethod_.empty() || ((! tmvaVariables_.empty() || ( !tmvaEtaVariables_.empty() )) && version == USER) );
  }

  if (( ! cutBased_ ) && (runMvas_)) {
    if (etaBinnedWeights_) {
      for (int v = 0; v < nEtaBins_; v++) {
        etaReader_.push_back(getMVA(tmvaEtaVariables_.at(v), tmvaEtaWeights.at(v), tmvaSpectators));
      }
    } else {
      reader_ = getMVA(tmvaVariables_, tmvaWeights, tmvaSpectators);
    }
  }
}

Member Function Documentation

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

inline

Definition at line 82 of file PileupJetIdAlgo.h.

References betaStarCut_, getMVA(), AlCaHLTBitMon_QueryRunRegistry::string, hltPUIdAlgo_cff::tmvaSpectators, and hltPUIdAlgo_cff::tmvaWeights.

{ return betaStarCut_; }

bool PileupJetIdAlgo::AlgoGBRForestsAndConstants::cutBased ( ) const

inline

Definition at line 68 of file PileupJetIdAlgo.h.

References cutBased_.

Referenced by PileupJetIdAlgo::runMva().

{ return cutBased_; }

bool PileupJetIdAlgo::AlgoGBRForestsAndConstants::etaBinnedWeights ( ) const

inline

Definition at line 69 of file PileupJetIdAlgo.h.

References etaBinnedWeights_.

Referenced by PileupJetIdAlgo::runMva().

{ return etaBinnedWeights_; }

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

inline

Definition at line 67 of file PileupJetIdAlgo.h.

References etaReader_.

Referenced by PileupJetIdAlgo::runMva().

{ return etaReader_; }

std::unique_ptr< const GBRForest > PileupJetIdAlgo::AlgoGBRForestsAndConstants::getMVA	(	std::vector< std::string > const &	varList,
		std::string const &	tmvaWeights,
		std::vector< std::string > const &	tmvaSpectators
	)

Definition at line 114 of file PileupJetIdAlgo.cc.

References patPFMETCorrections_cff::algo, relativeConstraints::empty, PileupJetIdAlgo::getVariables(), reco::details::loadTMVAWeights(), tmvaMethod_, and tmvaNames_.

Referenced by AlgoGBRForestsAndConstants(), and betaStarCut().

                                                                                                {

  // A temporary only to access the variables while calling TMVA AddVariable and TMVA AddSpectator.
  PileupJetIdAlgo algo(nullptr);

  TMVA::Reader tmpTMVAReader( "!Color:Silent:!Error" );
  for (auto const& varName : varList) {
    if ( tmvaNames_[varName].empty() ) tmvaNames_[varName] = varName;
    tmpTMVAReader.AddVariable( varName, std::get<float *,float>(algo.getVariables().at(tmvaNames_[varName])) );
  }
  for (auto const& spectatorName : tmvaSpectators) {
    if ( tmvaNames_[spectatorName].empty() ) tmvaNames_[spectatorName] = spectatorName;
    tmpTMVAReader.AddSpectator( spectatorName, std::get<float *,float>(algo.getVariables().at(tmvaNames_[spectatorName])) );
  }
  reco::details::loadTMVAWeights(&tmpTMVAReader, tmvaMethod_, tmvaWeights);
  return ( std::make_unique<const GBRForest> ( dynamic_cast<TMVA::MethodBDT*>( tmpTMVAReader.FindMVA(tmvaMethod_.c_str()) ) ) );
}

std::vector<double> const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::jEtaMax ( ) const

inline

Definition at line 73 of file PileupJetIdAlgo.h.

References jEtaMax_.

Referenced by PileupJetIdAlgo::runMva().

{ return jEtaMax_; }

std::vector<double> const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::jEtaMin ( ) const

inline

Definition at line 72 of file PileupJetIdAlgo.h.

References jEtaMin_.

Referenced by PileupJetIdAlgo::runMva().

{ return jEtaMin_; }

std::string const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::label ( ) const

inline

Definition at line 74 of file PileupJetIdAlgo.h.

References label_.

{ return label_; }

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

inline

Definition at line 80 of file PileupJetIdAlgo.h.

References mvacut_.

Referenced by PileupJetIdAlgo::computeIDflag().

{ return mvacut_; }

int PileupJetIdAlgo::AlgoGBRForestsAndConstants::nEtaBins ( ) const

inline

Definition at line 71 of file PileupJetIdAlgo.h.

References nEtaBins_.

Referenced by PileupJetIdAlgo::runMva().

{ return nEtaBins_; }

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

inline

Definition at line 66 of file PileupJetIdAlgo.h.

References reader_.

Referenced by PileupJetIdAlgo::runMva().

{ return reader_; }

array_t const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::rmsCut ( ) const

inline

Definition at line 81 of file PileupJetIdAlgo.h.

References rmsCut_.

{ return rmsCut_; }

bool PileupJetIdAlgo::AlgoGBRForestsAndConstants::runMvas ( ) const

inline

Definition at line 70 of file PileupJetIdAlgo.h.

References runMvas_.

Referenced by PileupJetIdAlgo::computeIdVariables().

{ return runMvas_; }

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

inline

Definition at line 77 of file PileupJetIdAlgo.h.

References tmvaEtaVariables_.

Referenced by PileupJetIdAlgo::runMva().

{ return tmvaEtaVariables_; }

std::string const& PileupJetIdAlgo::AlgoGBRForestsAndConstants::tmvaMethod ( ) const

inline

Definition at line 75 of file PileupJetIdAlgo.h.

References tmvaMethod_.

Referenced by PileupJetIdAlgo::method().

{ return tmvaMethod_; }

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

inline

Definition at line 76 of file PileupJetIdAlgo.h.

References tmvaVariables_.

Referenced by PileupJetIdAlgo::runMva().

{ return tmvaVariables_; }

Member Data Documentation

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

private

Definition at line 105 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and betaStarCut().

bool PileupJetIdAlgo::AlgoGBRForestsAndConstants::cutBased_

private

Definition at line 92 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and cutBased().

bool PileupJetIdAlgo::AlgoGBRForestsAndConstants::etaBinnedWeights_

private

Definition at line 93 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and etaBinnedWeights().

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

private

Definition at line 91 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and etaReader().

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

private

Definition at line 97 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and jEtaMax().

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

private

Definition at line 96 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and jEtaMin().

std::string PileupJetIdAlgo::AlgoGBRForestsAndConstants::label_

private

Definition at line 98 of file PileupJetIdAlgo.h.

Referenced by SequenceTypes.DummyModule::__repr__(), Mixins._Labelable::_findDependencies(), label(), and Mixins._Labelable::setLabel().

float PileupJetIdAlgo::AlgoGBRForestsAndConstants::mvacut_[3][4][4]

private

Definition at line 103 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and mvacut().

int PileupJetIdAlgo::AlgoGBRForestsAndConstants::nEtaBins_

private

Definition at line 95 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and nEtaBins().

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

private

Definition at line 90 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and reader().

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

private

Definition at line 104 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and rmsCut().

bool PileupJetIdAlgo::AlgoGBRForestsAndConstants::runMvas_

private

Definition at line 94 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and runMvas().

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

private

Definition at line 101 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and tmvaEtaVariables().

std::string PileupJetIdAlgo::AlgoGBRForestsAndConstants::tmvaMethod_

private

Definition at line 99 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), getMVA(), and tmvaMethod().

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

private

Definition at line 107 of file PileupJetIdAlgo.h.

Referenced by getMVA().

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

private

Definition at line 100 of file PileupJetIdAlgo.h.

Referenced by AlgoGBRForestsAndConstants(), and tmvaVariables().