RecurrentNetwork.h

Go to the documentation of this file.

#ifndef RECURRENTNETWORK_H
#define RECURRENTNETWORK_H

#include "Network.h"
#include "InputLayer.h"
#include "RecurrentNeuron.h"

namespace annie {

typedef TLayer<RecurrentNeuron> RecurrentLayer;
        

//GeneticRN(list<VECTOR> genotype);

class RecurrentNetwork : public Network {
public:
    static const int INPUT_NEURONS_LABEL_OFFSET;
    RecurrentNetwork(uint initialInputCount=0, uint initialRecurrenCount=0);

    void addInput(real initialValue=0);
    void addInput(InputNeuron *in) { _inputs.addNeuron(in); }

    void addNeuron(real bias=0, real activation=0);
    void addNeuron(RecurrentNeuron *n) { _neurons.addNeuron(n); }

    virtual const char* getClassName() const { return "RecurrentNetwork"; }

    virtual uint getInputCount() const { return _inputs.size(); }

    virtual uint getOutputCount() const { return _neurons.size(); }

    virtual void setInput(const Vector &input);

    RecurrentNeuron &getNeuron(uint index) { return _neurons[index]; }
    RecurrentNeuron &operator[](uint index) { return getNeuron(index); }
    
    InputNeuron &getInputNeuron(uint index) { return _inputs[index]; };

    virtual void resetActivations(const Vector &activations);

    virtual void step();
    
    virtual real getNeuronOutput(uint neuron) { return _neurons[neuron].getOutput(); }

    virtual Vector getOutput() const;

    virtual Vector getOutput(const Vector &input);

    virtual void save(const std::string &filename) { throw Exception("Not implemented"); }
   protected:
    InputLayer _inputs;
    RecurrentLayer _neurons;
    uint _time; //TODO: is it good for something?
};

} //namespace annie
#endif // _H

---