---

MultiLayerNetwork.h

Go to the documentation of this file.

#ifndef _MULTILAYERNETWORK_H
#define _MULTILAYERNETWORK_H


#include "Layer.h"
#include "InputLayer.h"
#include "Network.h"
#include "TrainingSet.h"

namespace annie
{
class MultiLayerNetwork : public Network
{
protected:
        int _nLayers;

        std::vector<Layer *> _layers;

        InputLayer *_inputLayer();

        Layer *_outputLayer();
public:
        MultiLayerNetwork(int inputs);

        MultiLayerNetwork(MultiLayerNetwork &srcNet);

        MultiLayerNetwork(const char *filename);

        virtual ~MultiLayerNetwork();

        virtual void addLayer(int size);

        virtual void connectLayer(int srcLayer);

        virtual void connect(int srcLayer, int srcNrn, int destNrn);

        virtual void connect(int srcLayer, int srcNrn, int destNrn, real weight);

        virtual void setBias(int layer, int nrn, real bias);

        virtual int getLayerCount();

        virtual VECTOR getOutput(VECTOR &input);

        virtual VECTOR getOutput(real *input);

        virtual void train(TrainingSet &T, int epochs, real learningRate, real momentum = 0.0);
        
        virtual void train(TrainingSet &T, int epochs, real learningRate, real momentum, bool verbose);

        virtual void save(const char *filename);

        virtual void setActivationFunction(int layer, ActivationFunction f, ActivationFunction df);

        virtual const char *getClassName();
};
}; //namespace annie
#endif // define _MULTILAYERNETWORK_H

---