PhotonDNNEstimator Class Reference

#include <PhotonDNNEstimator.h>

Public Member Functions
std::vector< 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	(	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

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

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

(

const reco::Photon &

ele

)

const

Definition at line 48 of file PhotonDNNEstimator.cc.

References reco::Photon::ecalPFClusterIso(), reco::Photon::ecalRecHitSumEtConeDR03(), reco::LeafCandidate::eta(), reco::Photon::full5x5_r9(), reco::Photon::full5x5_showerShapeVariables(), reco::Photon::full5x5_sigmaIetaIeta(), reco::Photon::hadTowOverEm(), reco::Photon::hadTowOverEmValid(), reco::Photon::hasPixelSeed(), reco::Photon::hcalPFClusterIso(), reco::Photon::hcalTowerSumEtConeDR03(), reco::LeafCandidate::pt(), reco::Photon::sigmaIetaIeta(), reco::Photon::ShowerShape::sigmaIetaIphi, reco::Photon::trkSumPtHollowConeDR03(), and L1TEGammaDiff_cfi::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;
}

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

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.

const egammaTools::EgammaDNNHelper PhotonDNNEstimator::dnnHelper_

private

Definition at line 36 of file PhotonDNNEstimator.h.

Referenced by evaluate(), and getSessions().

constexpr float PhotonDNNEstimator::ecalBarrelMaxEtaNoGap = 1.485

static

Definition at line 33 of file PhotonDNNEstimator.h.

constexpr float PhotonDNNEstimator::ecalBarrelMaxEtaWithGap = 1.566

static

Definition at line 32 of file PhotonDNNEstimator.h.

const bool PhotonDNNEstimator::useEBModelInGap_

private

Definition at line 38 of file PhotonDNNEstimator.h.