Loading [MathJax]/extensions/tex2jax.js
ATLAS Offline Software
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
MultifoldGNN.h
Go to the documentation of this file.
1 /*
2  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
3 */
4 
5 #ifndef MULTIFOLD_GNN_H
6 #define MULTIFOLD_GNN_H
7 
12 
13 #include <vector>
14 #include <string>
15 #include <memory>
16 #include <set>
17 
18 namespace FlavorTagInference {
19  class GNN;
20  struct GNNOptions;
21 }
22 
23 namespace FlavorTagInference {
24 
26  {
27  public:
28  MultifoldGNN(const std::vector<std::string>& folds,
29  const std::string& fold_hash_name,
31  MultifoldGNN(const std::vector<std::shared_ptr<const FlavorTagInference::GNN>>& folds,
32  const std::string& fold_hash_name);
36  void decorate(const xAOD::BTagging& btag) const;
37  void decorate(const xAOD::Jet& jet) const;
38  void decorateWithDefaults(const SG::AuxElement& jet) const;
39  void decorateWithDefaults(const xAOD::BTagging& btag) const;
40 
41  std::set<std::string> getDecoratorKeys() const;
42  std::set<std::string> getAuxInputKeys() const;
43  std::set<std::string> getConstituentAuxInputKeys() const;
44  private:
45  const FlavorTagInference::GNN& getFold(const SG::AuxElement& element) const;
46  std::vector<std::shared_ptr<const FlavorTagInference::GNN>> m_folds;
49  };
50 
51 }
52 #endif
FlavorTagInference::MultifoldGNN::decorate
void decorate(const xAOD::BTagging &btag) const
Definition: MultifoldGNN.cxx:58
GNN.h
FlavorTagInference::MultifoldGNN
Definition: MultifoldGNN.h:26
FlavorTagInference
This file contains "getter" functions used for accessing tagger inputs from the EDM.
Definition: AssociationEnums.h:11
FlavorTagInference::MultifoldGNN::MultifoldGNN
MultifoldGNN(MultifoldGNN &&)
SG::AuxElement
Base class for elements of a container that can have aux data.
Definition: AuxElement.h:483
FlavorTagInference::MultifoldGNN::getDecoratorKeys
std::set< std::string > getDecoratorKeys() const
Definition: MultifoldGNN.cxx:72
SG::ConstAccessor
Helper class to provide constant type-safe access to aux data.
Definition: ConstAccessor.h:55
FlavorTagInference::MultifoldGNN::MultifoldGNN
MultifoldGNN(const MultifoldGNN &)
GNNOptions.h
FlavorTagInference::MultifoldGNN::getConstituentAuxInputKeys
std::set< std::string > getConstituentAuxInputKeys() const
Definition: MultifoldGNN.cxx:78
JetContainerFwd.h
jet
Definition: JetCalibTools_PlotJESFactors.cxx:23
FlavorTagInference::MultifoldGNN::~MultifoldGNN
~MultifoldGNN()
FlavorTagInference::MultifoldGNN::m_jetLink
SG::AuxElement::ConstAccessor< ElementLink< xAOD::JetContainer > > m_jetLink
Definition: MultifoldGNN.h:48
xAOD::BTagging_v1
Definition: BTagging_v1.h:39
FlavorTagInference::MultifoldGNN::decorateWithDefaults
void decorateWithDefaults(const SG::AuxElement &jet) const
Definition: MultifoldGNN.cxx:64
FlavorTagInference::MultifoldGNN::MultifoldGNN
MultifoldGNN(const std::vector< std::string > &folds, const std::string &fold_hash_name, const FlavorTagInference::GNNOptions &opts)
Definition: MultifoldGNN.cxx:39
FlavorTagInference::GNNOptions
Definition: GNNOptions.h:16
python.FPGATrackSimAnalysisConfig.GNN
GNN
Definition: FPGATrackSimAnalysisConfig.py:697
xAOD::Jet_v1
Class describing a jet.
Definition: Jet_v1.h:57
FlavorTagInference::MultifoldGNN::getAuxInputKeys
std::set< std::string > getAuxInputKeys() const
Definition: MultifoldGNN.cxx:75
athena.opts
opts
Definition: athena.py:88
FlavorTagInference::MultifoldGNN::getFold
const FlavorTagInference::GNN & getFold(const SG::AuxElement &element) const
Definition: MultifoldGNN.cxx:82
FlavorTagInference::GNN
Definition: PhysicsAnalysis/JetTagging/FlavorTagInference/FlavorTagInference/GNN.h:39
FlavorTagInference::MultifoldGNN::m_folds
std::vector< std::shared_ptr< const FlavorTagInference::GNN > > m_folds
Definition: MultifoldGNN.h:46
BTaggingFwd.h
FlavorTagInference::MultifoldGNN::m_fold_hash
SG::AuxElement::ConstAccessor< uint32_t > m_fold_hash
Definition: MultifoldGNN.h:47