CacheData Class Reference

Public Member Functions
	CacheData (const edm::ParameterSet &conf)

Public Attributes
std::unique_ptr< const GBRForest >	haloTaggerGBR
std::unique_ptr< const PhotonDNNEstimator >	photonDNNEstimator

Detailed Description

Definition at line 57 of file GEDPhotonProducer.cc.

Constructor & Destructor Documentation

CacheData()

CacheData::CacheData ( const edm::ParameterSet &  conf )

inline

for MVA based beam halo tagger in the EE

Definition at line 59 of file GEDPhotonProducer.cc.

References DMR_cfg::config, createGBRForest(), edm::FileInPath::fullPath(), edm::ParameterSet::getParameter(), haloTaggerGBR, gedPhotons_cfi::mvaBasedHaloVariableSet, photonDNNEstimator, gedPhotons_cfi::runMVABasedHaloTagger, AlCaHLTBitMon_QueryRunRegistry::string, and particleFlowTmpBarrel_cfi::useEBModelInGap.

                                         {
    // Here we will have to load the DNN PFID if present in the config
    egammaTools::DNNConfiguration config;
    const auto& pset_dnn = conf.getParameter<edm::ParameterSet>("PhotonDNNPFid");
    const auto dnnEnabled = pset_dnn.getParameter<bool>("enabled");
    if (dnnEnabled) {
      config.inputTensorName = pset_dnn.getParameter<std::string>("inputTensorName");
      config.outputTensorName = pset_dnn.getParameter<std::string>("outputTensorName");
      config.modelsFiles = pset_dnn.getParameter<std::vector<std::string>>("modelsFiles");
      config.scalersFiles = pset_dnn.getParameter<std::vector<std::string>>("scalersFiles");
      config.outputDim = pset_dnn.getParameter<std::vector<unsigned int>>("outputDim");
      const auto useEBModelInGap = pset_dnn.getParameter<bool>("useEBModelInGap");
      photonDNNEstimator = std::make_unique<PhotonDNNEstimator>(config, useEBModelInGap);
    }
    const auto runMVABasedHaloTagger = conf.getParameter<bool>("runMVABasedHaloTagger");
    edm::ParameterSet mvaBasedHaloVariableSet = conf.getParameter<edm::ParameterSet>("mvaBasedHaloVariableSet");
    auto trainingFileName_ = mvaBasedHaloVariableSet.getParameter<edm::FileInPath>(("trainingFileName")).fullPath();
    if (runMVABasedHaloTagger) {
      haloTaggerGBR = createGBRForest(trainingFileName_);
    }
  }

Member Data Documentation

haloTaggerGBR

std::unique_ptr<const GBRForest> CacheData::haloTaggerGBR

Definition at line 82 of file GEDPhotonProducer.cc.

Referenced by CacheData().

photonDNNEstimator

std::unique_ptr<const PhotonDNNEstimator> CacheData::photonDNNEstimator

Definition at line 81 of file GEDPhotonProducer.cc.

Referenced by CacheData().