JSSWTopTaggerDNN.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JSSWTOPTAGGERDNN_H_
#define JSSWTOPTAGGERDNN_H_
 
#include "BoostedJetTaggers/JSSTaggerBase.h"
 
#include "lwtnn/LightweightNeuralNetwork.hh"
#include "lwtnn/parse_json.hh"
 
class JSSWTopTaggerDNN :
  public JSSTaggerBase {
    ASG_TOOL_CLASS0(JSSWTopTaggerDNN)
 
    public:
 
      JSSWTopTaggerDNN( const std::string& name );
 
      virtual StatusCode initialize() override;
 
      virtual StatusCode tag( const xAOD::Jet& jet ) const override;
 
    private:
 
      std::unique_ptr<lwt::LightweightNeuralNetwork> m_lwnn;
      std::map<std::string, double> m_DNN_inputValues;
 
      double getScore( const xAOD::Jet& jet ) const;
 
      std::map<std::string,double> getJetProperties( const xAOD::Jet& jet ) const;
 
  };
 
#endif