PhotonDNNEstimator Class Reference

#include <PhotonDNNEstimator.h>

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

Static Public Attributes
static const std::vector< std::string >	dnnAvaibleInputs
static constexpr float	ecalBarrelMaxEtaNoGap = 1.485
static constexpr float	ecalBarrelMaxEtaWithGap = 1.566

Private Attributes
const egammaTools::EgammaDNNHelper	dnnHelper_
const bool	useEBModelInGap_

Detailed Description

Definition at line 13 of file PhotonDNNEstimator.h.

Constructor & Destructor Documentation

PhotonDNNEstimator()

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

Definition at line 23 of file PhotonDNNEstimator.cc.

    : dnnHelper_(cfg,
                 std::bind(photonModelSelector,
                           _1,
                           (useEBModelInGap) ? PhotonDNNEstimator::ecalBarrelMaxEtaWithGap
                                             : PhotonDNNEstimator::ecalBarrelMaxEtaNoGap),
                 PhotonDNNEstimator::dnnAvaibleInputs),
      useEBModelInGap_(useEBModelInGap) {}

Member Function Documentation

evaluate()

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

Definition at line 69 of file PhotonDNNEstimator.cc.

References dnnHelper_, egammaTools::EgammaDNNHelper::evaluate(), getInputsVars(), PixelMapPlotter::inputs, displacedMuons_cfi::photon, and BPHMonitor_cfi::photons.

                                                                                              {
  // 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& photon : photons) {
    inputs.push_back(getInputsVars(photon));
  }
  return dnnHelper_.evaluate(inputs, sessions);
}

getInputsVars()

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

(

const reco::Photon &

ele

)

const

Definition at line 48 of file PhotonDNNEstimator.cc.

References displacedMuons_cfi::photon, and hcalRecHitTable_cff::variables.

Referenced by evaluate().

                                                                                         {
  // Prepare a map with all the defined variables
  std::map<std::string, float> variables;
  variables["pt"] = photon.pt();
  variables["eta"] = photon.eta();
  variables["hadTowOverEm"] = photon.hadTowOverEmValid() ? photon.hadTowOverEm() : 0;
  variables["trkSumPtHollowConeDR03"] = photon.trkSumPtHollowConeDR03();
  variables["EcalRecHit"] = photon.ecalRecHitSumEtConeDR03();
  variables["SigmaIetaIeta"] = photon.sigmaIetaIeta();
  variables["SigmaIetaIetaFull5x5"] = photon.full5x5_sigmaIetaIeta();
  variables["SigmaIEtaIPhiFull5x5"] = photon.full5x5_showerShapeVariables().sigmaIetaIphi;
  variables["EcalPFClusterIso"] = photon.ecalPFClusterIso();
  variables["HcalPFClusterIso"] = photon.hcalPFClusterIso();
  variables["HasPixelSeed"] = (Int_t)photon.hasPixelSeed();
  variables["R9Full5x5"] = photon.full5x5_r9();
  variables["hcalTower"] = photon.hcalTowerSumEtConeDR03();
  variables["R9Full5x5"] = photon.full5x5_r9();
  // Define more variables here and use them directly in the model config!
  return variables;
}

getSessions()

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

(

)

const

Definition at line 32 of file PhotonDNNEstimator.cc.

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

{ return dnnHelper_.getSessions(); };

Member Data Documentation

dnnAvaibleInputs

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

static

Initial value:

= {{"pt",
                                                                        "eta",
                                                                        "hadTowOverEm",
                                                                        "trkSumPtHollowConeDR03",
                                                                        "EcalRecHit",
                                                                        "SigmaIetaIeta",
                                                                        "SigmaIetaIetaFull5x5",
                                                                        "SigmaIEtaIPhiFull5x5",
                                                                        "EcalPFClusterIso",
                                                                        "HcalPFClusterIso",
                                                                        "HasPixelSeed",
                                                                        "R9Full5x5",
                                                                        "hcalTower"}}

Definition at line 30 of file PhotonDNNEstimator.h.

dnnHelper_

const egammaTools::EgammaDNNHelper PhotonDNNEstimator::dnnHelper_

private

Definition at line 36 of file PhotonDNNEstimator.h.

Referenced by evaluate(), and getSessions().

ecalBarrelMaxEtaNoGap

constexpr float PhotonDNNEstimator::ecalBarrelMaxEtaNoGap = 1.485

static

Definition at line 33 of file PhotonDNNEstimator.h.

ecalBarrelMaxEtaWithGap

constexpr float PhotonDNNEstimator::ecalBarrelMaxEtaWithGap = 1.566

static

Definition at line 32 of file PhotonDNNEstimator.h.

useEBModelInGap_

const bool PhotonDNNEstimator::useEBModelInGap_

private

Definition at line 38 of file PhotonDNNEstimator.h.