GRLMLPData.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 GRLMLPDATA_H
#define GRLMLPDATA_H
 
#include <TObject.h>
 
namespace Belle2 {
  class GRLMLPData : public TObject {
  public:
    GRLMLPData(): m_inputSamples(), m_targetSamples() { }
    ~GRLMLPData() { }
 
    void addSample(const std::vector<float>& input, const std::vector<float>& target)
    {
      m_inputSamples.push_back(input);
      m_targetSamples.push_back(target);
    }
 
    unsigned getNumberOfSamples() const { return m_targetSamples.size(); }
    const std::vector<float>& getInput(unsigned i) const { return m_inputSamples[i]; }
    const std::vector<float>& getTarget(unsigned i) const { return m_targetSamples[i]; }
 
  private:
    std::vector<std::vector<float>> m_inputSamples;
    std::vector<std::vector<float>> m_targetSamples;
 
    ClassDef(GRLMLPData, 1);
  };
}
#endif