DirectStorageDataSet.h

Go to the documentation of this file.

#ifndef OPENANN_IO_DIRECT_STORAGE_DATA_SET_H_
#define OPENANN_IO_DIRECT_STORAGE_DATA_SET_H_
#if __GNUC__ >= 4
#pragma GCC diagnostic ignored "-Wunused-parameter"
#endif

#include <OpenANN/io/DataSet.h>
#include <OpenANN/io/Logger.h>

class Stopwatch;

namespace OpenANN
{

class Evaluator;

class DirectStorageDataSet : public DataSet
{
protected:
  Eigen::MatrixXd* in;
  Eigen::MatrixXd* out;
  const int N;
  const int D;
  const int F;
  Eigen::VectorXd temporaryInput;
  Eigen::VectorXd temporaryOutput;
  Evaluator* evaluator; 

public:
  DirectStorageDataSet(Eigen::MatrixXd* in, Eigen::MatrixXd* out = 0,
                       Evaluator* evaluator = 0);
  virtual int samples() { return N; }
  virtual int inputs() { return D; }
  virtual int outputs() { return F; }
  virtual Eigen::VectorXd& getInstance(int i);
  virtual Eigen::VectorXd& getTarget(int i);
  virtual void finishIteration(Learner& learner);
};

} // namespace OpenANN

#endif // OPENANN_IO_DIRECT_STORAGE_DATA_SET_H_