---

backprop.h

Go to the documentation of this file.

#ifndef __BACKPROPTRAINER_H__
#define __BACKPROPTRAINER_H__

#include "trainer.h"


// ----              |                      -----           o                 //
// |   )  ___   ___  |    --            --    |        ___      _    ___      //
// |---   ___| |   \ | / |  ) |/\  __  |  )   |   |/\  ___| | |/ \  /   ) |/\ //
// |   ) (   | |     |/  |--  |   /  \ |--    |   |   (   | | |   | |---  |   //
// |___   \__|  \__/ | \ |    |   \__/ |      |   |    \__| | |   |  \__  |   //

class BackpropTrainer : public Trainer {
  public:
    virtual Array<DynParameter>*    parameters  () const;
    virtual void                    init        (const StringMap& params);
    
  protected:
    virtual void                    initTrain       (ANNetwork& network) const;

    virtual double                  trainOnce       (ANNetwork& network, const PatternSet& set) const;

    virtual double                  trainPattern    (ANNetwork& network, const PatternSet& set, int p) const;

    virtual void                    backpropagate   (ANNetwork& network, const PatternSet& set, int p) const;

    virtual void                    updateWeights   (ANNetwork& network) const;

  protected:
    double  mEta;

    double  mMomentum;

    double  mDecay;

    bool    mBatchLearning;

    mutable Vector  mWeightDeltas;

    mutable Vector  mError;
};

#endif

---