#include <ElectronDNNEstimator.h>

Public Member Functions
	ElectronDNNEstimator (const egammaTools::DNNConfiguration &, const bool useEBModelInGap)

std::vector< 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	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	(	const egammaTools::DNNConfiguration &	cfg,
		const bool	useEBModelInGap
	)

Definition at line 29 of file ElectronDNNEstimator.cc.

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

Member Function Documentation

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

Definition at line 144 of file ElectronDNNEstimator.cc.

References dnnHelper_, 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);
 }

std::map< std::string, float > ElectronDNNEstimator::getInputsVars ( const reco::GsfElectron & ele ) const

Definition at line 87 of file ElectronDNNEstimator.cc.

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