JSSWTopTaggerANN.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JSSWTOPTAGGERANN_H_
#define JSSWTOPTAGGERANN_H_
 
#include "BoostedJetTaggers/JSSTaggerBase.h"
 
#include "lwtnn/LightweightGraph.hh"
#include "lwtnn/parse_json.hh"
#include "lwtnn/Exceptions.hh"
#include "lwtnn/lightweight_nn_streamers.hh"
 
class JSSWTopTaggerANN :
  public JSSTaggerBase {
    ASG_TOOL_CLASS0(JSSWTopTaggerANN)
 
    public:
 
      JSSWTopTaggerANN(const std::string& name);
 
      virtual StatusCode initialize() override;
 
      virtual StatusCode tag(const xAOD::Jet& jet) const override;
      virtual StatusCode decorate(  const xAOD::JetContainer& jets  ) const override;
 
    private:
 
      std::unique_ptr<lwt::LightweightGraph> m_lwnn;
      std::map<std::string, std::map<std::string,double>> m_ANN_inputValues;   // variables for ANN
 
      std::vector<std::string> m_out_names;
 
      double getScore(const xAOD::Jet& jet) const;
 
      std::map<std::string, std::map<std::string,double>> getJetProperties(const xAOD::Jet& jet) const;
 
  };
 
#endif