GsfElectronAlgo::HeavyObjectCache Class Reference

#include <GsfElectronAlgo.h>

Public Member Functions
	HeavyObjectCache (const edm::ParameterSet &)

Public Attributes
std::unique_ptr< const ElectronDNNEstimator >	iElectronDNNEstimator
std::unique_ptr< const ElectronMVAEstimator >	iElectronMVAEstimator
std::unique_ptr< const SoftElectronMVAEstimator >	sElectronMVAEstimator

Detailed Description

Definition at line 56 of file GsfElectronAlgo.h.

Constructor & Destructor Documentation

GsfElectronAlgo::HeavyObjectCache::HeavyObjectCache

(

const edm::ParameterSet &

conf

)

Definition at line 39 of file GsfElectronAlgo.cc.

References edm::ParameterSet::getParameter(), AlCaHLTBitMon_QueryRunRegistry::string, parallelization::uint(), ElectronMVAEstimator::Configuration::vweightsfiles, and SoftElectronMVAEstimator::Configuration::vweightsfiles.

                                                                             {
  // soft electron MVA
  SoftElectronMVAEstimator::Configuration sconfig;
  sconfig.vweightsfiles = conf.getParameter<std::vector<std::string>>("SoftElecMVAFilesString");
  sElectronMVAEstimator = std::make_unique<SoftElectronMVAEstimator>(sconfig);
  // isolated electron MVA
  ElectronMVAEstimator::Configuration iconfig;
  iconfig.vweightsfiles = conf.getParameter<std::vector<std::string>>("ElecMVAFilesString");
  iElectronMVAEstimator = std::make_unique<ElectronMVAEstimator>(iconfig);

  // Here we will have to load the DNN PFID if present in the config
  egammaTools::DNNConfiguration dconfig;
  const auto& pset_dnn = conf.getParameter<edm::ParameterSet>("EleDNNPFid");
  const bool dnnEnabled = pset_dnn.getParameter<bool>("enabled");
  if (dnnEnabled) {
    dconfig.inputTensorName = pset_dnn.getParameter<std::string>("inputTensorName");
    dconfig.outputTensorName = pset_dnn.getParameter<std::string>("outputTensorName");
    dconfig.modelsFiles = pset_dnn.getParameter<std::vector<std::string>>("modelsFiles");
    dconfig.scalersFiles = pset_dnn.getParameter<std::vector<std::string>>("scalersFiles");
    dconfig.outputDim = pset_dnn.getParameter<uint>("outputDim");
    const auto useEBModelInGap = pset_dnn.getParameter<bool>("useEBModelInGap");
    iElectronDNNEstimator = std::make_unique<ElectronDNNEstimator>(dconfig, useEBModelInGap);
  }
}

Member Data Documentation

std::unique_ptr<const ElectronDNNEstimator> GsfElectronAlgo::HeavyObjectCache::iElectronDNNEstimator

Definition at line 61 of file GsfElectronAlgo.h.

std::unique_ptr<const ElectronMVAEstimator> GsfElectronAlgo::HeavyObjectCache::iElectronMVAEstimator

Definition at line 60 of file GsfElectronAlgo.h.

std::unique_ptr<const SoftElectronMVAEstimator> GsfElectronAlgo::HeavyObjectCache::sElectronMVAEstimator

Definition at line 59 of file GsfElectronAlgo.h.

Referenced by GsfElectronAlgo::createElectron().