ElectronDNNEstimator Class Reference

#include <ElectronDNNEstimator.h>

Public Member Functions
	ElectronDNNEstimator (const egammaTools::DNNConfiguration &, const bool useEBModelInGap)
std::vector< std::pair< uint, std::vector< float > > >	evaluate (const reco::GsfElectronCollection &ele, const std::vector< tensorflow::Session *> &sessions) const
std::map< std::string, float >	getInputsVars (const reco::GsfElectron &ele) const
std::vector< tensorflow::Session * >	getSessions () const

Static Public Attributes
static const std::vector< std::string >	dnnAvaibleInputs
static constexpr float	ecalBarrelMaxEtaNoGap = 1.485
static constexpr float	ecalBarrelMaxEtaWithGap = 1.566
static constexpr float	endcapBoundary = 2.5
static constexpr float	extEtaBoundary = 2.65
static constexpr float	ptThreshold = 10.

Private Attributes
const egammaTools::EgammaDNNHelper	dnnHelper_
const bool	useEBModelInGap_

Detailed Description

Definition at line 12 of file ElectronDNNEstimator.h.

Constructor & Destructor Documentation

ElectronDNNEstimator()

ElectronDNNEstimator::ElectronDNNEstimator	(	const egammaTools::DNNConfiguration &	cfg,
		const bool	useEBModelInGap
	)

Definition at line 38 of file ElectronDNNEstimator.cc.

    : dnnHelper_(cfg,
                 std::bind(electronModelSelector,
                           _1,
                           ElectronDNNEstimator::ptThreshold,
                           (useEBModelInGap) ? ElectronDNNEstimator::ecalBarrelMaxEtaWithGap
                                             : ElectronDNNEstimator::ecalBarrelMaxEtaNoGap,
                           ElectronDNNEstimator::endcapBoundary,
                           ElectronDNNEstimator::extEtaBoundary),
                 ElectronDNNEstimator::dnnAvaibleInputs),
      useEBModelInGap_(useEBModelInGap) {}

Member Function Documentation

evaluate()

std::vector< std::pair< uint, std::vector< float > > > ElectronDNNEstimator::evaluate	(	const reco::GsfElectronCollection &	ele,
		const std::vector< tensorflow::Session *> &	sessions
	)		const

Definition at line 159 of file ElectronDNNEstimator.cc.

References dnnHelper_, pwdgSkimBPark_cfi::electrons, egammaTools::EgammaDNNHelper::evaluate(), getInputsVars(), and PixelMapPlotter::inputs.

                                                                                                     {
  // Collect the map of variables for each candidate and call the dnnHelper
  // Scaling, model selection and running is performed in the helper
  std::vector<std::map<std::string, float>> inputs;
  for (const auto& ele : electrons) {
    inputs.push_back(getInputsVars(ele));
  }
  return dnnHelper_.evaluate(inputs, sessions);
}

getInputsVars()

std::map< std::string, float > ElectronDNNEstimator::getInputsVars

(

const reco::GsfElectron &

ele

)

const

Definition at line 100 of file ElectronDNNEstimator.cc.

References funct::abs(), reco::GsfElectron::closestCtfTrackNLayers(), reco::GsfElectron::closestCtfTrackNormChi2(), reco::GsfElectron::convVtxFitProb(), reco::GsfElectron::deltaEtaSeedClusterTrackAtCalo(), reco::GsfElectron::deltaEtaSuperClusterTrackAtVtx(), reco::GsfElectron::deltaPhiSuperClusterTrackAtVtx(), reco::GsfElectron::dr03EcalRecHitSumEt(), reco::GsfElectron::dr03HcalTowerSumEt(), reco::GsfElectron::dr03TkSumPt(), reco::GsfElectron::ecalEnergy(), reco::GsfElectron::ecalEnergyError(), reco::GsfElectron::ecalPFClusterIso(), reco::GsfElectron::eEleClusterOverPout(), reco::GsfElectron::eSuperClusterOverP(), reco::LeafCandidate::eta(), reco::GsfElectron::fbrem(), reco::GsfElectron::full5x5_e1x5(), reco::GsfElectron::full5x5_e2x5Max(), reco::GsfElectron::full5x5_e5x5(), reco::GsfElectron::full5x5_hcalOverEcal(), reco::GsfElectron::full5x5_hcalOverEcalValid(), reco::GsfElectron::full5x5_r9(), reco::GsfElectron::full5x5_sigmaIetaIeta(), reco::GsfElectron::full5x5_sigmaIphiIphi(), reco::GsfElectron::gsfTrack(), reco::GsfElectron::hadronicOverEm(), reco::GsfElectron::hcalOverEcalValid(), reco::GsfElectron::hcalPFClusterIso(), reco::HitPattern::MISSING_INNER_HITS, reco::GsfElectron::numberOfBrems(), reco::LeafCandidate::pt(), reco::GsfElectron::superCluster(), reco::GsfElectron::superClusterFbrem(), reco::GsfElectron::trackMomentumAtVtx(), reco::GsfElectron::trackMomentumError(), and EcalPhiSymFlatTableProducers_cfi::variables.

Referenced by evaluate().

                                                                                             {
  // Prepare a map with all the defined variables
  std::map<std::string, float> variables;
  variables["pt"] = ele.pt();
  variables["eta"] = ele.eta();
  variables["fbrem"] = ele.fbrem();
  variables["abs(deltaEtaSuperClusterTrackAtVtx)"] = std::abs(ele.deltaEtaSuperClusterTrackAtVtx());
  variables["abs(deltaPhiSuperClusterTrackAtVtx)"] = std::abs(ele.deltaPhiSuperClusterTrackAtVtx());
  variables["full5x5_sigmaIetaIeta"] = ele.full5x5_sigmaIetaIeta();
  variables["full5x5_hcalOverEcal"] = ele.full5x5_hcalOverEcalValid() ? ele.full5x5_hcalOverEcal() : 0;
  variables["eSuperClusterOverP"] = ele.eSuperClusterOverP();
  variables["full5x5_e1x5"] = ele.full5x5_e1x5();
  variables["eEleClusterOverPout"] = ele.eEleClusterOverPout();
  variables["closestCtfTrackNormChi2"] = ele.closestCtfTrackNormChi2();
  variables["closestCtfTrackNLayers"] = ele.closestCtfTrackNLayers();
  variables["gsfTrack.missing_inner_hits"] =
      ele.gsfTrack()->hitPattern().numberOfLostHits(reco::HitPattern::MISSING_INNER_HITS);
  variables["dr03TkSumPt"] = ele.dr03TkSumPt();
  variables["dr03EcalRecHitSumEt"] = ele.dr03EcalRecHitSumEt();
  variables["dr03HcalTowerSumEt"] = ele.dr03HcalTowerSumEt();
  variables["gsfTrack.normalizedChi2"] = ele.gsfTrack()->normalizedChi2();
  variables["superCluster.eta"] = ele.superCluster()->eta();
  variables["ecalPFClusterIso"] = ele.ecalPFClusterIso();
  variables["hcalPFClusterIso"] = ele.hcalPFClusterIso();
  variables["numberOfBrems"] = ele.numberOfBrems();
  variables["abs(deltaEtaSeedClusterTrackAtCalo)"] = std::abs(ele.deltaEtaSeedClusterTrackAtCalo());
  variables["hadronicOverEm"] = ele.hcalOverEcalValid() ? ele.hadronicOverEm() : 0;
  variables["full5x5_e2x5Max"] = ele.full5x5_e2x5Max();
  variables["full5x5_e5x5"] = ele.full5x5_e5x5();

  variables["full5x5_sigmaIphiIphi"] = ele.full5x5_sigmaIphiIphi();
  variables["1_minus_full5x5_e1x5_ratio_full5x5_e5x5"] = 1 - ele.full5x5_e1x5() / ele.full5x5_e5x5();
  variables["full5x5_e1x5_ratio_full5x5_e5x5"] = ele.full5x5_e1x5() / ele.full5x5_e5x5();
  variables["full5x5_r9"] = ele.full5x5_r9();
  variables["gsfTrack.trackerLayersWithMeasurement"] = ele.gsfTrack()->hitPattern().trackerLayersWithMeasurement();
  variables["gsfTrack.numberOfValidPixelBarrelHits"] = ele.gsfTrack()->hitPattern().numberOfValidPixelBarrelHits();
  variables["gsfTrack.numberOfValidPixelEndcapHits"] = ele.gsfTrack()->hitPattern().numberOfValidPixelEndcapHits();
  variables["superCluster.energy"] = ele.superCluster()->energy();
  variables["superCluster.rawEnergy"] = ele.superCluster()->rawEnergy();
  variables["superClusterFbrem"] = ele.superClusterFbrem();
  variables["1_ratio_ecalEnergy_minus_1_ratio_trackMomentumAtVtx.R"] =
      1 / ele.ecalEnergy() - 1 / ele.trackMomentumAtVtx().R();
  variables["superCluster.preshowerEnergy_ratio_superCluster.rawEnergy"] =
      ele.superCluster()->preshowerEnergy() / ele.superCluster()->rawEnergy();
  variables["convVtxFitProb"] = ele.convVtxFitProb();
  variables["superCluster.clustersSize"] = ele.superCluster()->clustersSize();
  variables["ecalEnergyError_ratio_ecalEnergy"] = ele.ecalEnergyError() / ele.ecalEnergy();
  variables["superClusterFbrem_plus_superCluster.energy"] = ele.superClusterFbrem() + ele.superCluster()->energy();
  variables["superClusterFbrem_plus_superCluster.rawEnergy"] =
      ele.superClusterFbrem() + ele.superCluster()->rawEnergy();
  variables["trackMomentumError"] = ele.trackMomentumError();
  variables["trackMomentumError_ratio_pt"] = ele.trackMomentumError() / ele.pt();
  variables["full5x5_e5x5_ratio_superCluster.rawEnergy"] = ele.full5x5_e5x5() / ele.superCluster()->rawEnergy();
  variables["full5x5_e5x5_ratio_superCluster.energy"] = ele.full5x5_e5x5() / ele.superCluster()->energy();

  // Define more variables here and use them directly in the model config!
  return variables;
}

getSessions()

std::vector< tensorflow::Session * > ElectronDNNEstimator::getSessions

(

)

const

Definition at line 50 of file ElectronDNNEstimator.cc.

References dnnHelper_, and egammaTools::EgammaDNNHelper::getSessions().

{ return dnnHelper_.getSessions(); };

Member Data Documentation

dnnAvaibleInputs

const std::vector< std::string > ElectronDNNEstimator::dnnAvaibleInputs

static

Definition at line 29 of file ElectronDNNEstimator.h.

dnnHelper_

const egammaTools::EgammaDNNHelper ElectronDNNEstimator::dnnHelper_

private

Definition at line 38 of file ElectronDNNEstimator.h.

Referenced by evaluate(), and getSessions().

ecalBarrelMaxEtaNoGap

constexpr float ElectronDNNEstimator::ecalBarrelMaxEtaNoGap = 1.485

static

Definition at line 33 of file ElectronDNNEstimator.h.

ecalBarrelMaxEtaWithGap

constexpr float ElectronDNNEstimator::ecalBarrelMaxEtaWithGap = 1.566

static

Definition at line 32 of file ElectronDNNEstimator.h.

endcapBoundary

constexpr float ElectronDNNEstimator::endcapBoundary = 2.5

static

Definition at line 34 of file ElectronDNNEstimator.h.

extEtaBoundary

constexpr float ElectronDNNEstimator::extEtaBoundary = 2.65

static

Definition at line 35 of file ElectronDNNEstimator.h.

ptThreshold

constexpr float ElectronDNNEstimator::ptThreshold = 10.

static

Definition at line 31 of file ElectronDNNEstimator.h.

useEBModelInGap_

const bool ElectronDNNEstimator::useEBModelInGap_

private

Definition at line 40 of file ElectronDNNEstimator.h.