GRLNeuro.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#ifndef GRLNEURO_H
#define GRLNEURO_H
 
#include <trg/grl/dataobjects/GRLMLP.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <framework/database/DBObjPtr.h>
//#include <trg/cdc/dbobjects/CDCTriggerNeuroConfig.h>
//#include <framework/dataobjects/BinnedEventT0.h>
 
namespace Belle2 {
  class GRLNeuro {
  public:
    struct Parameters {
      unsigned nMLP;
      std::vector<std::vector<float>> nHidden;
      bool targetresult = true;
      bool multiplyHidden = false;
      std::vector<std::vector<float>> outputScale = {{ -1., 1.}};
 
      unsigned n_cdc_sector;
      std::vector<float> i_cdc_sector;
      unsigned n_ecl_sector;
      std::vector<float> i_ecl_sector;
    };
 
 
 
    GRLNeuro() {}
 
    virtual ~GRLNeuro() {}
 
    void initialize(const Parameters& p);
    //void initialize(const std::vector<float>& nodes);
 
    GRLMLP& operator[](unsigned index) { return m_MLPs[index]; }
    const GRLMLP& operator[](unsigned index) const { return m_MLPs[index]; }
    unsigned nSectors() const { return m_MLPs.size(); }
 
    float relu(float x);
 
    float float_to_fixed(float num, int m, int n);
 
    float mysigmiod(float num);
    void save(const std::string& filename, const std::string& arrayname = "MLPs");
    bool load(unsigned isector, const std::string& wfilename, const std::string& bfilename);
 
    float runMLP(unsigned isector,  const std::vector<float>& input);
 
  private:
    std::vector<GRLMLP> m_MLPs = {};
 
  };
}
#endif