HGCalTriggerClusterIdentificationBDT Class Reference

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 8 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 44 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 70 of file HGCalTriggerClusterIdentificationBDT.cc.

    : HGCalTriggerClusterIdentificationBase() {}

HGCalTriggerClusterIdentificationBDT::~HGCalTriggerClusterIdentificationBDT ( )

inlineoverride

Definition at line 38 of file HGCalTriggerClusterIdentificationBDT.cc.

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

{};

Member Function Documentation

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

private

Definition at line 159 of file HGCalTriggerClusterIdentificationBDT.cc.

References eostools::cat(), categories_, 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 167 of file HGCalTriggerClusterIdentificationBDT.cc.

References cl3d_coreshowerlength, cl3d_firstlayer, cl3d_maxlayer, cl3d_seemax, cl3d_seetot, cl3d_showerlength, cl3d_sppmax, cl3d_spptot, cl3d_srrmax, cl3d_srrmean, cl3d_srrtot, cl3d_szz, l1t::HGCalClusterT< C >::coreShowerLength(), l1t::HGCalClusterT< C >::firstLayer(), l1t::HGCalClusterT< C >::maxLayer(), l1t::HGCalClusterT< C >::showerLength(), l1t::HGCalClusterT< C >::sigmaEtaEtaMax(), l1t::HGCalClusterT< C >::sigmaEtaEtaTot(), l1t::HGCalClusterT< C >::sigmaPhiPhiMax(), l1t::HGCalClusterT< C >::sigmaPhiPhiTot(), l1t::HGCalClusterT< C >::sigmaRRMax(), l1t::HGCalClusterT< C >::sigmaRRMean(), l1t::HGCalClusterT< C >::sigmaRRTot(), and l1t::HGCalClusterT< C >::sigmaZZ().

Referenced by 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 151 of file HGCalTriggerClusterIdentificationBDT.cc.

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

Referenced by ~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 73 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, taus_updatedMVAIds_cff::variable, and working_points_.

Referenced by ~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 140 of file HGCalTriggerClusterIdentificationBDT.cc.

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

Referenced by average.Average::average(), decision(), 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.);
}

Member Data Documentation

std::vector<std::unique_ptr<TMVAEvaluator> > HGCalTriggerClusterIdentificationBDT::bdts_

private

Definition at line 59 of file HGCalTriggerClusterIdentificationBDT.cc.

Referenced by initialize(), and value().

std::vector<Category> HGCalTriggerClusterIdentificationBDT::categories_

private

Definition at line 58 of file HGCalTriggerClusterIdentificationBDT.cc.

Referenced by category(), and initialize().

std::vector<std::string> HGCalTriggerClusterIdentificationBDT::input_variables_

private

Definition at line 61 of file HGCalTriggerClusterIdentificationBDT.cc.

Referenced by initialize(), and value().

std::vector<ClusterVariable> HGCalTriggerClusterIdentificationBDT::input_variables_id_

private

Definition at line 62 of file HGCalTriggerClusterIdentificationBDT.cc.

Referenced by initialize(), and value().

std::vector<double> HGCalTriggerClusterIdentificationBDT::working_points_

private

Definition at line 60 of file HGCalTriggerClusterIdentificationBDT.cc.

Referenced by decision(), and initialize().