#include <TauNNId.h>

Public Member Functions
float	compute (const l1t::PFCandidate &iSeed, l1t::PFCandidateCollection &iParts)

float	EvaluateNN ()

void	setNNVectorVar ()

	TauNNId (const std::string &iInput, const TauNNTFCache *cache, const std::string &iWeightFile, int iNParticles)

	~TauNNId ()

Private Attributes
unique_ptr< float[]>	fEta_

unique_ptr< float[]>	fId_

std::string	fInput_

int	fNParticles_

unique_ptr< float[]>	fPhi_

unique_ptr< float[]>	fPt_

std::vector< float >	NNvectorVar_

tensorflow::Session *	session_

Detailed Description

Definition at line 13 of file TauNNId.h.

Constructor & Destructor Documentation

◆ TauNNId()

TauNNId::TauNNId	(	const std::string &	iInput,
		const TauNNTFCache *	cache,
		const std::string &	iWeightFile,
		int	iNParticles
	)

Definition at line 8 of file TauNNId.cc.

                                                                                                                   {
   NNvectorVar_.clear();
   edm::FileInPath fp(iWeightFile);
   session_ = tensorflow::createSession(cache->graphDef);
   fNParticles_ = iNParticles;
  
   fPt_ = std::make_unique<float[]>(fNParticles_);
   fEta_ = std::make_unique<float[]>(fNParticles_);
   fPhi_ = std::make_unique<float[]>(fNParticles_);
   fId_ = std::make_unique<float[]>(fNParticles_);
   fInput_ = iInput;
 }

References utilities::cache(), tensorflow::createSession(), fEta_, fId_, fInput_, fNParticles_, personalPlayback::fp, fPhi_, fPt_, NNvectorVar_, and session_.

◆ ~TauNNId()

TauNNId::~TauNNId ( )

Definition at line 21 of file TauNNId.cc.

21 { tensorflow::closeSession(session_); }

References tensorflow::closeSession(), and session_.

Member Function Documentation

◆ compute()

float TauNNId::compute	(	const l1t::PFCandidate &	iSeed,
		l1t::PFCandidateCollection &	iParts
	)

Definition at line 51 of file TauNNId.cc.

                                                                                     {
   for (int i0 = 0; i0 < fNParticles_; i0++) {
     fPt_.get()[i0] = 0;
     fEta_.get()[i0] = 0;
     fPhi_.get()[i0] = 0;
     fId_.get()[i0] = 0;
   }
   std::sort(iParts.begin(), iParts.end(), [](l1t::PFCandidate i, l1t::PFCandidate j) { return (i.pt() > j.pt()); });
   for (unsigned int i0 = 0; i0 < iParts.size(); i0++) {
     if (i0 > n_particles_max || i0 >= (unsigned int)fNParticles_)
       break;
     fPt_.get()[i0] = iParts[i0].pt();
     fEta_.get()[i0] = iSeed.eta() - iParts[i0].eta();
     fPhi_.get()[i0] = deltaPhi(iSeed.phi(), iParts[i0].phi());
     fId_.get()[i0] = iParts[i0].id();
   }
   setNNVectorVar();
   return EvaluateNN();
 }

References SiPixelRawToDigiRegional_cfi::deltaPhi, reco::LeafCandidate::eta(), EvaluateNN(), fEta_, fId_, fNParticles_, fPhi_, fPt_, mps_fire::i, dqmiolumiharvest::j, n_particles_max, reco::LeafCandidate::phi(), setNNVectorVar(), and jetUpdater_cfi::sort.

◆ EvaluateNN()

float TauNNId::EvaluateNN ( )

Definition at line 40 of file TauNNId.cc.

                           {
   tensorflow::Tensor input(tensorflow::DT_FLOAT,
                            {1, (unsigned int)NNvectorVar_.size()});  //was {1,35} but get size mismatch, CHECK
   for (unsigned int i = 0; i < NNvectorVar_.size(); i++) {
     input.matrix<float>()(0, i) = float(NNvectorVar_[i]);
   }
   std::vector<tensorflow::Tensor> outputs;
   tensorflow::run(session_, {{fInput_, input}}, {"dense_4/Sigmoid:0"}, &outputs);
   return outputs[0].matrix<float>()(0, 0);
 }  //end EvaluateNN

References fInput_, dqmMemoryStats::float, mps_fire::i, input, createfilelist::int, NNvectorVar_, PatBasicFWLiteJetAnalyzer_Selector_cfg::outputs, tensorflow::run(), and session_.

Referenced by compute().

◆ setNNVectorVar()

void TauNNId::setNNVectorVar ( )

Definition at line 22 of file TauNNId.cc.

                              {
   NNvectorVar_.clear();
   for (int i0 = 0; i0 < fNParticles_; i0++) {
     NNvectorVar_.push_back(fPt_.get()[i0]);   //pT
     NNvectorVar_.push_back(fEta_.get()[i0]);  //dEta from jet axis
     NNvectorVar_.push_back(fPhi_.get()[i0]);  //dPhi from jet axis
     if (fPt_.get()[i0] == 0) {
       for (int i1 = 0; i1 < 5; i1++)
         NNvectorVar_.push_back(0);
       continue;
     }
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::Photon);         // Photon
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::Electron);       // Electron
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::Muon);           // Muon
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::NeutralHadron);  // Neutral Had
     NNvectorVar_.push_back(fId_.get()[i0] == l1t::PFCandidate::ChargedHadron);  // Charged Had
   }
 }