HGCalTriggerClusterIdentificationBDT Class Reference

Inheritance diagram for HGCalTriggerClusterIdentificationBDT:

Classes
class	Category
class	WorkingPoint

Public Member Functions
bool	decision (const l1t::HGCalMulticluster &cluster, unsigned wp=0) const final
	HGCalTriggerClusterIdentificationBDT ()
void	initialize (const edm::ParameterSet &conf) final
float	value (const l1t::HGCalMulticluster &cluster) const final
const std::vector< std::string > &	working_points () 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< std::vector< WorkingPoint > >	working_points_
std::vector< std::string >	working_points_names_

Detailed Description

Definition at line 8 of file HGCalTriggerClusterIdentificationBDT.cc.

Member Enumeration Documentation

ClusterVariable

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 58 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::HGCalTriggerClusterIdentificationBDT

(

)

Definition at line 85 of file HGCalTriggerClusterIdentificationBDT.cc.

    : HGCalTriggerClusterIdentificationBase() {}

~HGCalTriggerClusterIdentificationBDT()

HGCalTriggerClusterIdentificationBDT::~HGCalTriggerClusterIdentificationBDT ( )

inlineoverride

Definition at line 51 of file HGCalTriggerClusterIdentificationBDT.cc.

{};

Member Function Documentation

category()

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

private

Definition at line 189 of file HGCalTriggerClusterIdentificationBDT.cc.

References eostools::cat(), categories_, edm::contains(), PVValHelper::eta, and DiDispStaMuonMonitor_cfi::pt.

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;
}

clusterVariable()

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

private

Definition at line 197 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(), l1t::HGCalClusterT< C >::sigmaZZ(), and HPSPFTaus_cff::variable.

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.;
}

decision()

bool HGCalTriggerClusterIdentificationBDT::decision ( const l1t::HGCalMulticluster &  cluster, unsigned  wp = 0  ) const

finalvirtual

Implements HGCalTriggerClusterIdentificationBase.

Definition at line 181 of file HGCalTriggerClusterIdentificationBDT.cc.

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

                                                                                                          {
  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).at(wp).working_point() : true);
}

initialize()

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

finalvirtual

Implements HGCalTriggerClusterIdentificationBase.

Definition at line 88 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, geometryDiff::file, contentValuesFiles::fullPath, edm::ParameterSet::getParameter(), input_variables_, input_variables_id_, candidateCombinedMVAV2Computer_cfi::spectators, AlCaHLTBitMon_QueryRunRegistry::string, HPSPFTaus_cff::variable, working_points_, and working_points_names_.

                                                                                 {
  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");

  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()) {
    throw cms::Exception("HGCalTriggerClusterIdentificationBDT|BadInitialization")
        << "Inconsistent numbers of categories, BDT weight files and working points";
  }
  size_t categories_size = categories_etamin.size();

  const auto wps_conf = conf.getParameter<std::vector<edm::ParameterSet>>("WorkingPoints");
  working_points_.resize(categories_size);
  for (const auto& wp_conf : wps_conf) {
    std::string wp_name = wp_conf.getParameter<std::string>("Name");
    std::vector<double> wps = wp_conf.getParameter<std::vector<double>>("WorkingPoint");
    working_points_names_.emplace_back(wp_name);
    if (wps.size() != categories_size) {
      throw cms::Exception("HGCalTriggerClusterIdentificationBDT|BadInitialization")
          << "Inconsistent number of categories in working point '" << wp_name << "'";
    }
    for (size_t cat = 0; cat < categories_size; cat++) {
      working_points_[cat].emplace_back(wp_name, wps[cat]);
    }
  }

  categories_.reserve(categories_size);
  bdts_.reserve(categories_size);
  for (size_t cat = 0; cat < categories_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);
  }
}

value()

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

finalvirtual

Implements HGCalTriggerClusterIdentificationBase.

Definition at line 170 of file HGCalTriggerClusterIdentificationBDT.cc.

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

Referenced by average.Average::average(), and decision().

                                                                                           {
  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.);
}

working_points()

const std::vector<std::string>& HGCalTriggerClusterIdentificationBDT::working_points ( ) const

inlinefinalvirtual

Implements HGCalTriggerClusterIdentificationBase.

Definition at line 55 of file HGCalTriggerClusterIdentificationBDT.cc.

References working_points_names_.

{ return working_points_names_; }

Member Data Documentation

bdts_

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

private

Definition at line 73 of file HGCalTriggerClusterIdentificationBDT.cc.

Referenced by initialize(), and value().

categories_

std::vector<Category> HGCalTriggerClusterIdentificationBDT::categories_

private

Definition at line 72 of file HGCalTriggerClusterIdentificationBDT.cc.

Referenced by category(), and initialize().

input_variables_

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

private

Definition at line 75 of file HGCalTriggerClusterIdentificationBDT.cc.

Referenced by initialize(), and value().

input_variables_id_

std::vector<ClusterVariable> HGCalTriggerClusterIdentificationBDT::input_variables_id_

private

Definition at line 76 of file HGCalTriggerClusterIdentificationBDT.cc.

Referenced by initialize(), and value().

working_points_

std::vector<std::vector<WorkingPoint> > HGCalTriggerClusterIdentificationBDT::working_points_

private

Definition at line 74 of file HGCalTriggerClusterIdentificationBDT.cc.

Referenced by decision(), and initialize().

working_points_names_

std::vector<std::string> HGCalTriggerClusterIdentificationBDT::working_points_names_

private

Definition at line 77 of file HGCalTriggerClusterIdentificationBDT.cc.

Referenced by initialize(), and working_points().