GNNDataLoader.h

Go to the documentation of this file.

/*
Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#pragma once
#include "FlavorTagInference/GNNOptions.h"
#include "FlavorTagInference/DataPrepUtilities.h"
#include "FlavorTagInference/SaltModelEDMLoaderBase.h"
#include "FlavorTagInference/DataPrepUtilities.h"
 
namespace FlavorTagInference {
 
    class GNNDataLoader : public SaltModelEDMLoaderBase {
    public:
        GNNDataLoader(std::shared_ptr<const SaltModel> salt_model, const GNNOptions& opts);
        FTagOptions ftag_options;
        FTagDataDependencyNames data_dependency_names;
    private:
        GNNOptions m_gnn_options;
        std::string getVecInputName(const SaltModelVersion salt_model_version, const ConstituentsInputConfig& constituents_config) const;
    };
} // namespace FlavorTagInference