RadialBasisNetwork.h

Go to the documentation of this file.

#ifndef _RADIALBASISNETWORK_H
#define _RADIALBASISNETWORK_H

#include "Network.h"
#include "SimpleNeuron.h"
#include "CenterNeuron.h"
#include "InputLayer.h"
#include "TrainingSet.h"

namespace annie
{

class RadialBasisNetwork : public Network
{
protected:
    int _nCenters;

    InputLayer *_inputLayer;

    Layer *_centerLayer;

    Layer *_outputLayer;

public:
    RadialBasisNetwork(int inputs, int centers, int outputs);

    RadialBasisNetwork(const char *filename);

    RadialBasisNetwork(RadialBasisNetwork &srcNet);

    virtual ~RadialBasisNetwork();

    virtual void setCenter(uint i, Vector &center);

    virtual void setCenter(uint i, real *center);

    virtual Vector getCenter(int i) const;

    virtual uint getCenterCount() const;

    virtual void setBias(uint i, real bias);

    virtual real getBias(uint i) const;

    virtual void removeBias(uint i);

    virtual void setWeight(int center, int output, real weight);

    virtual real getWeight(int center, int output) const;

    virtual Vector getOutput(Vector &input);

    virtual Vector getOutput(real *input);

    void setCenterActivationFunction(ActivationFunction f, ActivationFunction df);

    void trainWeights(TrainingSet &T);
    //void trainCentersAndWeights(TrainingSet &T, int epochs, real learningRate);

    virtual const char *getClassName() const;

    virtual void save(const char *filename);
};
}; //namespace annie
#endif // define _RADIALBASISNETWORK_H

---