TauNNId Class Reference

#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	(	const std::string &	iInput,
		const TauNNTFCache *	cache,
		const std::string &	iWeightFile,
		int	iNParticles
	)

Definition at line 8 of file TauNNId.cc.

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

                                                                                                                  {
  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;
}

TauNNId::~TauNNId

(

)

Definition at line 21 of file TauNNId.cc.

References tensorflow::closeSession(), and session_.

{ tensorflow::closeSession(session_); }

Member Function Documentation

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

Definition at line 51 of file TauNNId.cc.

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

                                                                                    {
  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();
}

float TauNNId::EvaluateNN

(

)

Definition at line 40 of file TauNNId.cc.

References fInput_, mps_fire::i, input, NNvectorVar_, tensorflow::run(), and session_.

Referenced by compute().

                          {
  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

void TauNNId::setNNVectorVar

(

)

Definition at line 22 of file TauNNId.cc.

References l1t::PFCandidate::ChargedHadron, l1t::PFCandidate::Electron, fEta_, fId_, fNParticles_, fPhi_, fPt_, l1t::PFCandidate::Muon, l1t::PFCandidate::NeutralHadron, NNvectorVar_, and l1t::PFCandidate::Photon.

Referenced by compute().

                             {
  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
  }
}

Member Data Documentation

unique_ptr<float[]> TauNNId::fEta_

private

Definition at line 28 of file TauNNId.h.

Referenced by compute(), setNNVectorVar(), and TauNNId().

unique_ptr<float[]> TauNNId::fId_

private

Definition at line 30 of file TauNNId.h.

Referenced by compute(), setNNVectorVar(), and TauNNId().

std::string TauNNId::fInput_

private

Definition at line 25 of file TauNNId.h.

Referenced by EvaluateNN(), and TauNNId().

int TauNNId::fNParticles_

private

Definition at line 26 of file TauNNId.h.

Referenced by compute(), setNNVectorVar(), and TauNNId().

unique_ptr<float[]> TauNNId::fPhi_

private

Definition at line 29 of file TauNNId.h.

Referenced by compute(), setNNVectorVar(), and TauNNId().

unique_ptr<float[]> TauNNId::fPt_

private

Definition at line 27 of file TauNNId.h.

Referenced by compute(), setNNVectorVar(), and TauNNId().

std::vector<float> TauNNId::NNvectorVar_

private

Definition at line 24 of file TauNNId.h.

Referenced by EvaluateNN(), setNNVectorVar(), and TauNNId().

tensorflow::Session* TauNNId::session_

private

Definition at line 23 of file TauNNId.h.

Referenced by EvaluateNN(), TauNNId(), and ~TauNNId().