Inheritance diagram for HGCalTriggerClusterIdentificationBDT:

Classes
class	Category

Public Member Functions
bool	decision (const l1t::HGCalMulticluster &cluster) const final

	HGCalTriggerClusterIdentificationBDT ()

void	initialize (const edm::ParameterSet &conf) final

float	value (const l1t::HGCalMulticluster &cluster) const final

	~HGCalTriggerClusterIdentificationBDT () override

Public Member Functions inherited from HGCalTriggerClusterIdentificationBase
	HGCalTriggerClusterIdentificationBase ()

virtual	~HGCalTriggerClusterIdentificationBase ()

Private Types
enum	ClusterVariable { ClusterVariable::cl3d_showerlength, ClusterVariable::cl3d_coreshowerlength, ClusterVariable::cl3d_firstlayer, ClusterVariable::cl3d_maxlayer, ClusterVariable::cl3d_seetot, ClusterVariable::cl3d_seemax, ClusterVariable::cl3d_spptot, ClusterVariable::cl3d_sppmax, ClusterVariable::cl3d_szz, ClusterVariable::cl3d_srrtot, ClusterVariable::cl3d_srrmax, ClusterVariable::cl3d_srrmean }

Private Member Functions
int	category (float pt, float eta) const

float	clusterVariable (ClusterVariable, const l1t::HGCalMulticluster &) const

Private Attributes
std::vector< std::unique_ptr< TMVAEvaluator > >	bdts_

std::vector< Category >	categories_

std::vector< std::string >	input_variables_

std::vector< ClusterVariable >	input_variables_id_

std::vector< double >	working_points_

Detailed Description

Definition at line 10 of file HGCalTriggerClusterIdentificationBDT.cc.

Member Enumeration Documentation

enum HGCalTriggerClusterIdentificationBDT::ClusterVariable

strongprivate

Enumerator
cl3d_showerlength
cl3d_coreshowerlength
cl3d_firstlayer
cl3d_maxlayer
cl3d_seetot
cl3d_seemax
cl3d_spptot
cl3d_sppmax
cl3d_szz
cl3d_srrtot
cl3d_srrmax
cl3d_srrmean

Definition at line 52 of file HGCalTriggerClusterIdentificationBDT.cc.

     {
       cl3d_showerlength,
       cl3d_coreshowerlength,
       cl3d_firstlayer,
       cl3d_maxlayer,
       cl3d_seetot,
       cl3d_seemax,
       cl3d_spptot,
       cl3d_sppmax,
       cl3d_szz,
       cl3d_srrtot,
       cl3d_srrmax,
       cl3d_srrmean
     };

Constructor & Destructor Documentation

HGCalTriggerClusterIdentificationBDT::HGCalTriggerClusterIdentificationBDT ( )

Definition at line 83 of file HGCalTriggerClusterIdentificationBDT.cc.

References initialize().

                                       :HGCalTriggerClusterIdentificationBase()
 {
 }

HGCalTriggerClusterIdentificationBDT::~HGCalTriggerClusterIdentificationBDT ( )

inlineoverride

Definition at line 46 of file HGCalTriggerClusterIdentificationBDT.cc.

References decision(), initialize(), and value().

46 {};

Member Function Documentation

int HGCalTriggerClusterIdentificationBDT::category	(	float	pt,
		float	eta
	)		const

private

Definition at line 186 of file HGCalTriggerClusterIdentificationBDT.cc.

References eostools::cat(), categories_, clusterVariable(), and edm::contains().

Referenced by decision(), and value().

 {
   for(unsigned cat=0; cat<categories_.size(); cat++)
   {
     if(categories_[cat].contains(pt, eta)) return static_cast<int>(cat);
   }
   return -1;
 }

float HGCalTriggerClusterIdentificationBDT::clusterVariable	(	ClusterVariable	variable,
		const l1t::HGCalMulticluster &	cluster
	)		const

private

Definition at line 200 of file HGCalTriggerClusterIdentificationBDT.cc.

Referenced by category(), and value().

 {
   switch(variable)
   {
     case ClusterVariable::cl3d_showerlength: return cluster.showerLength();
     case ClusterVariable::cl3d_coreshowerlength: return cluster.coreShowerLength();
     case ClusterVariable::cl3d_firstlayer: return cluster.firstLayer();
     case ClusterVariable::cl3d_maxlayer: return cluster.maxLayer();
     case ClusterVariable::cl3d_seetot: return cluster.sigmaEtaEtaTot();
     case ClusterVariable::cl3d_seemax: return cluster.sigmaEtaEtaMax();
     case ClusterVariable::cl3d_spptot: return cluster.sigmaPhiPhiTot();
     case ClusterVariable::cl3d_sppmax: return cluster.sigmaPhiPhiMax();
     case ClusterVariable::cl3d_szz: return cluster.sigmaZZ();
     case ClusterVariable::cl3d_srrtot: return cluster.sigmaRRTot();
     case ClusterVariable::cl3d_srrmax: return cluster.sigmaRRMax();
     case ClusterVariable::cl3d_srrmean: return cluster.sigmaRRMean();
     default: break;
   }
   return 0.;
 }

bool HGCalTriggerClusterIdentificationBDT::decision ( const l1t::HGCalMulticluster & cluster ) const

finalvirtual

Implements HGCalTriggerClusterIdentificationBase.

Definition at line 175 of file HGCalTriggerClusterIdentificationBDT.cc.

References eostools::cat(), category(), PVValHelper::eta, reco::LeafCandidate::eta(), EnergyCorrector::pt, reco::LeafCandidate::pt(), value(), and working_points_.

Referenced by value(), and ~HGCalTriggerClusterIdentificationBDT().

 {
   float bdt_output = value(cluster);
   float pt = cluster.pt();
   float eta = cluster.eta();
   int cat = category(pt, eta);
   return (cat!=-1 ? bdt_output>working_points_.at(cat) : true);
 }

void HGCalTriggerClusterIdentificationBDT::initialize ( const edm::ParameterSet & conf )

finalvirtual

Implements HGCalTriggerClusterIdentificationBase.

Definition at line 89 of file HGCalTriggerClusterIdentificationBDT.cc.

References bdts_, eostools::cat(), categories_, cl3d_coreshowerlength, cl3d_firstlayer, cl3d_maxlayer, cl3d_seemax, cl3d_seetot, cl3d_showerlength, cl3d_sppmax, cl3d_spptot, cl3d_srrmax, cl3d_srrmean, cl3d_srrtot, cl3d_szz, Exception, FrontierConditions_GlobalTag_cff::file, edm::ParameterSet::getParameter(), input_variables_, input_variables_id_, candidateCombinedMVAV2Computer_cfi::spectators, value(), taus_updatedMVAIds_cff::variable, and working_points_.

Referenced by HGCalTriggerClusterIdentificationBDT(), and ~HGCalTriggerClusterIdentificationBDT().

 {
   if(!bdts_.empty())
   {
     edm::LogWarning("HGCalTriggerClusterIdentificationBDT|Initialization")
       << "BDTs already initialized.";
     return; 
   }
   input_variables_ = conf.getParameter< std::vector<std::string> >("Inputs");
   std::vector<std::string> bdt_files = conf.getParameter< std::vector<std::string> >("Weights");
   std::vector<double> categories_etamin = conf.getParameter<std::vector<double>>("CategoriesEtaMin");
   std::vector<double> categories_etamax = conf.getParameter<std::vector<double>>("CategoriesEtaMax");
   std::vector<double> categories_ptmin = conf.getParameter<std::vector<double>>("CategoriesPtMin");
   std::vector<double> categories_ptmax = conf.getParameter<std::vector<double>>("CategoriesPtMax");
   working_points_ = conf.getParameter<std::vector<double>>("WorkingPoints");
 
   if(bdt_files.size()!=categories_etamin.size() ||
       categories_etamin.size()!=categories_etamax.size() ||
       categories_etamax.size()!=categories_ptmin.size() ||
       categories_ptmin.size()!=categories_ptmax.size() ||
       categories_ptmax.size()!=working_points_.size()
     )
   {
     throw cms::Exception("HGCalTriggerClusterIdentificationBDT|BadInitialization")
       << "Inconsistent numbers of categories, BDT weight files and working points";
   }
   categories_.reserve(working_points_.size());
   bdts_.reserve(working_points_.size());
   for(unsigned cat=0; cat<categories_etamin.size(); cat++)
   {
     categories_.emplace_back(
         categories_ptmin[cat],
         categories_ptmax[cat],
         categories_etamin[cat],
         categories_etamax[cat]);
   }
   std::vector<std::string> spectators = {};
   for (const auto& file : bdt_files)
   {
     bdts_.emplace_back(new TMVAEvaluator());
     bdts_.back()->initialize(
         "!Color:Silent:!Error",
         "BDT::bdt",
         edm::FileInPath(file).fullPath(),
         input_variables_,
         spectators,
         false, false);
   }
 
   // Transform input variable strings to enum values for later comparisons
   input_variables_id_.reserve(input_variables_.size());
   for(const auto& variable : input_variables_)
   {
     if(variable=="cl3d_showerlength") input_variables_id_.push_back(ClusterVariable::cl3d_showerlength);
     else if(variable=="cl3d_coreshowerlength") input_variables_id_.push_back(ClusterVariable::cl3d_coreshowerlength);
     else if(variable=="cl3d_firstlayer") input_variables_id_.push_back(ClusterVariable::cl3d_firstlayer);
     else if(variable=="cl3d_maxlayer") input_variables_id_.push_back(ClusterVariable::cl3d_maxlayer);
     else if(variable=="cl3d_seetot") input_variables_id_.push_back(ClusterVariable::cl3d_seetot);
     else if(variable=="cl3d_seemax") input_variables_id_.push_back(ClusterVariable::cl3d_seemax);
     else if(variable=="cl3d_spptot") input_variables_id_.push_back(ClusterVariable::cl3d_spptot);
     else if(variable=="cl3d_sppmax") input_variables_id_.push_back(ClusterVariable::cl3d_sppmax);
     else if(variable=="cl3d_szz") input_variables_id_.push_back(ClusterVariable::cl3d_szz);
     else if(variable=="cl3d_srrtot") input_variables_id_.push_back(ClusterVariable::cl3d_srrtot);
     else if(variable=="cl3d_srrmax") input_variables_id_.push_back(ClusterVariable::cl3d_srrmax);
     else if(variable=="cl3d_srrmean") input_variables_id_.push_back(ClusterVariable::cl3d_srrmean);
   }
 }

float HGCalTriggerClusterIdentificationBDT::value ( const l1t::HGCalMulticluster & cluster ) const

finalvirtual

Implements HGCalTriggerClusterIdentificationBase.

Definition at line 159 of file HGCalTriggerClusterIdentificationBDT.cc.

References bdts_, eostools::cat(), category(), clusterVariable(), decision(), PVValHelper::eta, reco::LeafCandidate::eta(), mps_fire::i, input_variables_, input_variables_id_, PatBasicFWLiteJetAnalyzer_Selector_cfg::inputs, EnergyCorrector::pt, and reco::LeafCandidate::pt().

Referenced by average.Average::average(), decision(), initialize(), and ~HGCalTriggerClusterIdentificationBDT().

 {
   std::map<std::string, float> inputs;
   for(unsigned i=0; i<input_variables_.size(); i++)
   {
     inputs[input_variables_[i]] = clusterVariable(input_variables_id_[i], cluster);
   }
   float pt = cluster.pt();
   float eta = cluster.eta();
   int cat = category(pt, eta);
   return (cat!=-1 ? bdts_.at(cat)->evaluate(inputs) : -999.);
 }