File indexing completed on 2023-03-17 11:13:12
0001 #ifndef L1TRIGGER_PHASE2L1PARTICLEFLOWS_TAUNNID_H
0002 #define L1TRIGGER_PHASE2L1PARTICLEFLOWS_TAUNNID_H
0003
0004 #include <string>
0005 #include "PhysicsTools/TensorFlow/interface/TensorFlow.h"
0006 #include "DataFormats/L1TParticleFlow/interface/PFCandidate.h"
0007
0008 class TauNNId {
0009 public:
0010 TauNNId(const std::string &iInput,
0011 const tensorflow::Session *session,
0012 const std::string &iWeightFile,
0013 int iNParticles);
0014 ~TauNNId(){};
0015
0016 void setNNVectorVar();
0017 float EvaluateNN();
0018 float compute(const l1t::PFCandidate &iSeed, l1t::PFCandidateCollection &iParts);
0019
0020 private:
0021 const tensorflow::Session *session_;
0022 std::vector<float> NNvectorVar_;
0023 std::string fInput_;
0024 int fNParticles_;
0025 unique_ptr<float[]> fPt_;
0026 unique_ptr<float[]> fEta_;
0027 unique_ptr<float[]> fPhi_;
0028 unique_ptr<float[]> fId_;
0029 };
0030 #endif
