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);

    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(); }

    void save(const std::string& filename, const std::string& arrayname = "MLPs");
    bool load(unsigned isector, const std::string& wfilename, const std::string& bfilename);
    bool load(unsigned isector, std::vector<float> weight, std::vector<float> bias);

    float runMLP(unsigned isector,  const std::vector<float>& input);
 
  private:
    std::vector<GRLMLP> m_MLPs = {};
 
  };

}
#endif