train.h

Go to the documentation of this file.

/***************************************************************************/
#ifndef AURESERVOIR_TRAIN_H__
#define AURESERVOIR_TRAIN_H__

#include "utilities.h"
#include "delaysum.h"

namespace aureservoir
{

enum TrainAlgorithm
{
  TRAIN_PI,        
  TRAIN_LS,        
  TRAIN_RIDGEREG,  
  TRAIN_DS_PI      
};

template <typename T> class ESN;

template <typename T>
class TrainBase
{
 public:

  TrainBase(ESN<T> *esn) { esn_=esn; }

  virtual ~TrainBase() {}

  virtual void train(const typename ESN<T>::DEMatrix &in,
                     const typename ESN<T>::DEMatrix &out,
                     int washout) throw(AUExcept) = 0;

 protected:

  void checkParams(const typename ESN<T>::DEMatrix &in,
                   const typename ESN<T>::DEMatrix &out,
                   int washout) throw(AUExcept);


  void collectStates(const typename ESN<T>::DEMatrix &in,
                     const typename ESN<T>::DEMatrix &out,
                     int washout);

  void squareStates();

  void clearData()
  { M.resize(1,1); O.resize(1,1); }

  ESN<T> *esn_;

  typename ESN<T>::DEMatrix M;
  typename ESN<T>::DEMatrix O;
};

template <typename T>
class TrainPI : public TrainBase<T>
{
  using TrainBase<T>::esn_;
  using TrainBase<T>::M;
  using TrainBase<T>::O;

 public:
  TrainPI(ESN<T> *esn) : TrainBase<T>(esn) {}
  virtual ~TrainPI() {}

  virtual void train(const typename ESN<T>::DEMatrix &in,
                     const typename ESN<T>::DEMatrix &out,
                     int washout) throw(AUExcept);
};

template <typename T>
class TrainLS : public TrainBase<T>
{
  using TrainBase<T>::esn_;
  using TrainBase<T>::M;
  using TrainBase<T>::O;

 public:
  TrainLS(ESN<T> *esn) : TrainBase<T>(esn) {}
  virtual ~TrainLS() {}

  virtual void train(const typename ESN<T>::DEMatrix &in,
                     const typename ESN<T>::DEMatrix &out,
                     int washout) throw(AUExcept);
};

template <typename T>
class TrainRidgeReg : public TrainBase<T>
{
  using TrainBase<T>::esn_;
  using TrainBase<T>::M;
  using TrainBase<T>::O;

 public:
  TrainRidgeReg(ESN<T> *esn) : TrainBase<T>(esn) {}
  virtual ~TrainRidgeReg() {}

  virtual void train(const typename ESN<T>::DEMatrix &in,
                     const typename ESN<T>::DEMatrix &out,
                     int washout) throw(AUExcept);
};

template <typename T>
class TrainDSPI : public TrainBase<T>
{
  using TrainBase<T>::esn_;
  using TrainBase<T>::M;
  using TrainBase<T>::O;

 public:
  TrainDSPI(ESN<T> *esn) : TrainBase<T>(esn) {}
  virtual ~TrainDSPI() {}

  virtual void train(const typename ESN<T>::DEMatrix &in,
                     const typename ESN<T>::DEMatrix &out,
                     int washout) throw(AUExcept);
};

} // end of namespace aureservoir

#endif // AURESERVOIR_TRAIN_H__

---