InputPreprocessor.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INPUT_PREPROCESSOR_HH_TAURECTOOLS
#define INPUT_PREPROCESSOR_HH_TAURECTOOLS
 
#include "tauRecTools/lwtnn/lightweight_network_config.h"
#include "tauRecTools/lwtnn/Exceptions.h"
 
#include <Eigen/Dense>
#include <map>
#include <vector>
 
namespace lwtDev {
 
  using Eigen::VectorXd;
  using Eigen::MatrixXd;
 
  // use a normal map externally, since these are more common in user
  // code.  TODO: is it worth changing to unordered_map?
  typedef std::map<std::string, double> ValueMap;
  typedef std::vector<std::pair<std::string, double> > ValueVector;
  typedef std::map<std::string, std::vector<double> > VectorMap;
 
  // ______________________________________________________________________
  // input preprocessor (handles normalization and packing into Eigen)
 
  class InputPreprocessor
  {
  public:
    InputPreprocessor(const std::vector<Input>& inputs);
    VectorXd operator()(const ValueMap&) const;
  private:
    // input transformations
    VectorXd m_offsets;
    VectorXd m_scales;
    std::vector<std::string> m_names;
  };
 
  class InputVectorPreprocessor
  {
  public:
    InputVectorPreprocessor(const std::vector<Input>& inputs);
    MatrixXd operator()(const VectorMap&) const;
  private:
    // input transformations
    VectorXd m_offsets;
    VectorXd m_scales;
    std::vector<std::string> m_names;
  };
}
 
#endif // INPUT_PREPROCESSOR_HH_TAURECTOOLS