GRLMLP.h

#ifndef GRLMLP_H
#define GRLMLP_H
 
#include <TObject.h>
#include <framework/logging/Logger.h>
 
namespace Belle2 {
  class GRLMLP : public TObject {
 
    // weights etc. are set only by the trainer
    friend class GRLNeuroTrainerModule;
 
  public:
    GRLMLP();
 
    GRLMLP(std::vector<unsigned short>& nodes, unsigned short targets, const std::vector<float>& outputscale);
 
    ~GRLMLP() { }
 
    bool isTrained() const { return trained; }
    unsigned nLayers() const { return nNodes.size(); }
    unsigned nNodesLayer(unsigned iLayer) const { return nNodes[iLayer]; }
    unsigned nWeights() const { return weights.size(); }
    unsigned nWeightsCal() const;
    std::vector<float> getWeights() const { return weights; }
 
  private:
    std::vector<unsigned short> nNodes;
    std::vector<float> weights;
    bool trained;
 
    unsigned short targetVars;
    std::vector<float> outputScale;
 
    ClassDef(GRLMLP, 2);
  };
}
#endif