FFNet.h

// Copyright (C) 2001 Jean-Marc Valin

#ifndef FFNET_H
#define FFNET_H

#include "Object.h"
#include "FFLayer.h"
#include <vector>
#include <iostream>
#include "Array.h"

namespace FD {

class TrainingAlgo;


class FFNet : public Object {
  protected:

   Vector<int> topo;

   Vector<RCPtr<FFLayer> > layers;

   float *weights;
   
   int nbNeurons;

   int nbWeights;

  public:
   FFNet(const Vector<int> &_topo, const Vector<std::string> &functions);
   //FFNet(const Vector<int> &_topo);
   FFNet() {}

   FFNet(FFNet &net);

   FFNet(const Vector<int> &_topo, const Vector<std::string> &functions, std::vector<float *> &tin, std::vector<float *> &tout);

   void init(const Vector<std::string> &functions);

   void setupLayersAfterRead();

   float *calc(const float *input, float *value, float *deriv=NULL)
   {
      layers[0]->update(input, value, deriv);
      for (int i=1;i<layers.size();i++)
      {
         if (deriv)
         {
            layers[i]->update(value+layers[i-1]->getNeuronOffset(), 
                              value+layers[i]->getNeuronOffset(), 
                              deriv+layers[i]->getNeuronOffset());
         } else {
            layers[i]->update(value+layers[i-1]->getNeuronOffset(), 
                              value+layers[i]->getNeuronOffset());
         }
      }
      return value+layers[layers.size()-1]->getNeuronOffset();
   }
  
   void learn(float *input, float *output, double *gradient, double *err=NULL, float *calc_output=NULL);

   void calcGradient(std::vector<float *> &tin, std::vector<float *> &tout, Array<float> weights, 
                     Array<double> &gradient, double &err);

   void weightedLearn(float *input, float *output, float *learnWeights, double *gradient, 
                      double *err=NULL, float *calc_output=NULL);

   void weightedCalcGradient(std::vector<float *> &tin, std::vector<float *> &tout, std::vector<float *> &learnWeights, 
                             Array<float> weights, Array<double> &gradient, double &err);

   float totalError(std::vector<float *> tin, std::vector<float *> tout);

   int getNbWeights () {return nbWeights;}

   int getNbNeurons () {return nbNeurons;}

   const Vector<int> &getTopo() {return topo;}

   const Vector<RCPtr<FFLayer> > &getLayers() {return layers;}

   const float *getWeights() {return weights;}

   void setWeights(float *ptr) {vec_copy(ptr, weights, nbWeights);}

   void setDerivOffset(float d);
   
   void printOn(std::ostream &out) const;

   void readFrom (std::istream &in);

};

std::istream &operator >> (std::istream &in, FFNet &net);

}//namespace FD
#endif

---