Belle II Software  release-05-01-25
GRLMLPData.h
1 #ifndef GRLMLPDATA_H
2 #define GRLMLPDATA_H
3 
4 #include <TObject.h>
5 
6 namespace Belle2 {
12  class GRLMLPData : public TObject {
13  public:
18 
20  void addSample(const std::vector<float>& input, const std::vector<float>& target)
21  {
22  inputSamples.push_back(input);
23  targetSamples.push_back(target);
24  }
25 
27  unsigned nSamples() const { return targetSamples.size(); }
29  const std::vector<float>& getInput(unsigned i) const { return inputSamples[i]; }
31  const std::vector<float>& getTarget(unsigned i) const { return targetSamples[i]; }
32 
33  private:
35  std::vector<std::vector<float>> inputSamples;
37  std::vector<std::vector<float>> targetSamples;
38 
40  ClassDef(GRLMLPData, 1);
41  };
43 }
44 #endif
Belle2::GRLMLPData::ClassDef
ClassDef(GRLMLPData, 1)
Needed to make the ROOT object storable.
Belle2::GRLMLPData
Struct for training data of a single MLP for the neuro trigger.
Definition: GRLMLPData.h:12
Belle2::GRLMLPData::getTarget
const std::vector< float > & getTarget(unsigned i) const
get target value of sample i
Definition: GRLMLPData.h:31
Belle2::GRLMLPData::nSamples
unsigned nSamples() const
get number of samples (same for input and target)
Definition: GRLMLPData.h:27
Belle2::GRLMLPData::inputSamples
std::vector< std::vector< float > > inputSamples
list of input vectors for network training.
Definition: GRLMLPData.h:35
Belle2
Abstract base class for different kinds of events.
Definition: MillepedeAlgorithm.h:19
Belle2::GRLMLPData::targetSamples
std::vector< std::vector< float > > targetSamples
list of target values for network training.
Definition: GRLMLPData.h:37
Belle2::GRLMLPData::GRLMLPData
GRLMLPData()
default constructor.
Definition: GRLMLPData.h:15
Belle2::GRLMLPData::~GRLMLPData
~GRLMLPData()
destructor, empty because we don't allocate memory anywhere.
Definition: GRLMLPData.h:17
Belle2::GRLMLPData::getInput
const std::vector< float > & getInput(unsigned i) const
get input vector of sample i
Definition: GRLMLPData.h:29
Belle2::GRLMLPData::addSample
void addSample(const std::vector< float > &input, const std::vector< float > &target)
add a pair of input and target
Definition: GRLMLPData.h:20