ElectronDNNCalculator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// Dear emacs, this is -*-c++-*-
 
#ifndef ELECTRONDNNCALCULATOR_H
#define ELECTRONDNNCALCULATOR_H
 
// This include is needed at the top before any includes regarding lwtnn
// since it includes Eigen in a specific way which causes compilation errors
// if not included before lwtnn
#include "EventPrimitives/EventPrimitives.h"
 
#include "AsgMessaging/AsgMessagingForward.h"
#include "ElectronPhotonSelectorTools/AsgElectronSelectorTool.h"
#include "TTree.h"
#include <vector>
#include <string>
#include "lwtnn/generic/FastGraph.hh"
 
 
class ElectronDNNCalculator : public asg::AsgMessagingForward
{
public:
  ElectronDNNCalculator( AsgElectronSelectorTool* owner,
                         const std::string& modelFileName,
                         const std::string& quantileFileName,
                         const std::vector<std::string>& variablesName,
                         const bool multiClass);

  ~ElectronDNNCalculator() {};

  Eigen::Matrix<float, -1, 1> calculate( const std::vector<double>& ) const;
 
private:
  double transformInput( const std::vector<double>& quantiles, double value ) const;
  int readQuantileTransformer( TTree* tree );

  std::unique_ptr<lwt::generic::FastGraph<float>> m_graph = 0;

  std::vector<std::vector<double>> m_quantiles;
  std::vector<double> m_references;
  bool m_multiClass;
  std::vector<std::string> m_variables;
  uint m_var_size;
};
 
#endif