CMS 3D CMS Logo

List of all members | Public Member Functions | Public Attributes
reco::SCProducerCache Class Reference

#include <SCProducerCache.h>

Public Member Functions

 SCProducerCache (const edm::ParameterSet &conf)
 

Public Attributes

reco::DeepSCConfiguration config
 
std::unique_ptr< const DeepSCGraphEvaluationdeepSCEvaluator
 

Detailed Description

Definition at line 9 of file SCProducerCache.h.

Constructor & Destructor Documentation

◆ SCProducerCache()

reco::SCProducerCache::SCProducerCache ( const edm::ParameterSet conf)
inline

Definition at line 12 of file SCProducerCache.h.

References config, deepSCEvaluator, edm::ParameterSet::getParameter(), AlCaHLTBitMon_QueryRunRegistry::string, and parallelization::uint.

12  {
13  // Here we will have to load the DNN PFID if present in the config
14  auto clustering_type = conf.getParameter<std::string>("ClusteringType");
15 
16  if (clustering_type == "DeepSC") {
17  const auto& pset_dnn = conf.getParameter<edm::ParameterSet>("deepSuperClusterConfig");
18  config.modelFile = pset_dnn.getParameter<std::string>("modelFile");
19  config.configFileClusterFeatures = pset_dnn.getParameter<std::string>("configFileClusterFeatures");
20  config.configFileWindowFeatures = pset_dnn.getParameter<std::string>("configFileWindowFeatures");
21  config.configFileHitsFeatures = pset_dnn.getParameter<std::string>("configFileHitsFeatures");
22  config.nClusterFeatures = pset_dnn.getParameter<uint>("nClusterFeatures");
23  config.nWindowFeatures = pset_dnn.getParameter<uint>("nWindowFeatures");
24  config.nHitsFeatures = pset_dnn.getParameter<uint>("nHitsFeatures");
25  config.maxNClusters = pset_dnn.getParameter<uint>("maxNClusters");
26  config.maxNRechits = pset_dnn.getParameter<uint>("maxNRechits");
27  config.batchSize = pset_dnn.getParameter<uint>("batchSize");
28  config.collectionStrategy = pset_dnn.getParameter<std::string>("collectionStrategy");
29  deepSCEvaluator = std::make_unique<DeepSCGraphEvaluation>(config);
30  }
31  };
reco::DeepSCConfiguration config
T getParameter(std::string const &) const
Definition: ParameterSet.h:303
Definition: config.py:1
std::unique_ptr< const DeepSCGraphEvaluation > deepSCEvaluator

Member Data Documentation

◆ config

reco::DeepSCConfiguration reco::SCProducerCache::config

◆ deepSCEvaluator

std::unique_ptr<const DeepSCGraphEvaluation> reco::SCProducerCache::deepSCEvaluator