FlavorTagDiscriminants::MultifoldGNN Class Reference

#include <MultifoldGNN.h>

Collaboration diagram for FlavorTagDiscriminants::MultifoldGNN:

Public Member Functions
	MultifoldGNN (const std::vector< std::string > &folds, const std::string &fold_hash_name, const GNNOptions &opts)
	MultifoldGNN (const std::vector< std::shared_ptr< const GNN >> &folds, const std::string &fold_hash_name)
	MultifoldGNN (MultifoldGNN &&)
	MultifoldGNN (const MultifoldGNN &)
	~MultifoldGNN ()
void	decorate (const xAOD::BTagging &btag) const
void	decorate (const xAOD::Jet &jet) const
void	decorateWithDefaults (const SG::AuxElement &jet) const
std::set< std::string >	getDecoratorKeys () const
std::set< std::string >	getAuxInputKeys () const
std::set< std::string >	getConstituentAuxInputKeys () const

Private Member Functions
const GNN &	getFold (const SG::AuxElement &element) const

Private Attributes
std::vector< std::shared_ptr< const GNN > >	m_folds
SG::AuxElement::ConstAccessor< uint32_t >	m_fold_hash
SG::AuxElement::ConstAccessor< ElementLink< xAOD::JetContainer > >	m_jetLink

Detailed Description

Definition at line 21 of file MultifoldGNN.h.

Constructor & Destructor Documentation

MultifoldGNN() [1/4]

FlavorTagDiscriminants::MultifoldGNN::MultifoldGNN	(	const std::vector< std::string > &	folds,
		const std::string &	fold_hash_name,
		const GNNOptions &	opts
	)

Definition at line 38 of file MultifoldGNN.cxx.

                        :
    MultifoldGNN(getNNs(nn_files, o), fold_hash_name)
  {
  }

MultifoldGNN() [2/4]

FlavorTagDiscriminants::MultifoldGNN::MultifoldGNN	(	const std::vector< std::shared_ptr< const GNN >> &	folds,
		const std::string &	fold_hash_name
	)

Definition at line 45 of file MultifoldGNN.cxx.

                                    :
    m_folds(nns),
    m_fold_hash(fold_hash_name),
    m_jetLink(jetLinkName)
  {
  }

MultifoldGNN() [3/4]

FlavorTagDiscriminants::MultifoldGNN::MultifoldGNN ( MultifoldGNN &&  )

default

MultifoldGNN() [4/4]

FlavorTagDiscriminants::MultifoldGNN::MultifoldGNN ( const MultifoldGNN &  )

default

~MultifoldGNN()

FlavorTagDiscriminants::MultifoldGNN::~MultifoldGNN ( )

default

Member Function Documentation

decorate() [1/2]

void FlavorTagDiscriminants::MultifoldGNN::decorate

(

const xAOD::BTagging &

btag

)

const

Definition at line 57 of file MultifoldGNN.cxx.

                                                            {
    getFold(**m_jetLink(btag)).decorate(btag);
  }

decorate() [2/2]

void FlavorTagDiscriminants::MultifoldGNN::decorate

(

const xAOD::Jet &

jet

)

const

Definition at line 60 of file MultifoldGNN.cxx.

                                                      {
    getFold(jet).decorate(jet);
  }

decorateWithDefaults()

void FlavorTagDiscriminants::MultifoldGNN::decorateWithDefaults

(

const SG::AuxElement &

jet

)

const

Definition at line 63 of file MultifoldGNN.cxx.

                                                                       {
    // note that the default values should be identical betwen all folds
    m_folds.at(0)->decorateWithDefaults(jet);
  }

getAuxInputKeys()

std::set< std::string > FlavorTagDiscriminants::MultifoldGNN::getAuxInputKeys

(

)

const

Definition at line 72 of file MultifoldGNN.cxx.

                                                        {
    return merged([](const auto& f){ return f.getAuxInputKeys(); }, m_folds);
  }

getConstituentAuxInputKeys()

std::set< std::string > FlavorTagDiscriminants::MultifoldGNN::getConstituentAuxInputKeys

(

)

const

Definition at line 75 of file MultifoldGNN.cxx.

                                                                   {
    return merged([](const auto& f){ return f.getConstituentAuxInputKeys(); }, m_folds);
  }

getDecoratorKeys()

std::set< std::string > FlavorTagDiscriminants::MultifoldGNN::getDecoratorKeys

(

)

const

Definition at line 69 of file MultifoldGNN.cxx.

                                                         {
    return merged([](const auto& f){ return f.getDecoratorKeys(); }, m_folds);
  }

getFold()

const GNN & FlavorTagDiscriminants::MultifoldGNN::getFold ( const SG::AuxElement &  element ) const

private

Definition at line 79 of file MultifoldGNN.cxx.

                                                                    {
    return *m_folds.at(m_fold_hash(element) % m_folds.size());
  }

Member Data Documentation

m_fold_hash

SG::AuxElement::ConstAccessor<uint32_t> FlavorTagDiscriminants::MultifoldGNN::m_fold_hash

private

Definition at line 42 of file MultifoldGNN.h.

m_folds

std::vector<std::shared_ptr<const GNN> > FlavorTagDiscriminants::MultifoldGNN::m_folds

private

Definition at line 41 of file MultifoldGNN.h.

m_jetLink

SG::AuxElement::ConstAccessor<ElementLink<xAOD::JetContainer> > FlavorTagDiscriminants::MultifoldGNN::m_jetLink

private

Definition at line 43 of file MultifoldGNN.h.

---

The documentation for this class was generated from the following files:

• MultifoldGNN.h
• MultifoldGNN.cxx