DataStream.h

Go to the documentation of this file.

#ifndef OPENANN_IO_DATA_STREAM_H_
#define OPENANN_IO_DATA_STREAM_H_

#include <Eigen/Core>

namespace OpenANN
{

class DirectStorageDataSet;
class Optimizer;
class Learner;

class DataStream
{
  int cacheSize, collected;
  Eigen::MatrixXd X, T;
  DirectStorageDataSet* cache;
  Optimizer* opt; 
  Learner* learner; 
public:
  DataStream(int cacheSize);
  ~DataStream();
  DataStream& setLearner(Learner& learner);
  DataStream& setOptimizer(Optimizer& opt);
  void addSample(Eigen::VectorXd* x, Eigen::VectorXd* t = 0);
private:
  void initialize(int inputs, int outputs);
};

} // namespace OpenANN

#endif // OPENANN_IO_DATA_STREAM_H_