NeuralNetwork/FeedforwardNeuralNetwork/FeedforwardThreeLayerNN.h

Go to the documentation of this file.

/*
    <one line to give the program's name and a brief idea of what it does.>
    Copyright (C) <year>  <name of author>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.

*/

#ifndef FEEDFORWARDTHREELAYERNN_H
#define FEEDFORWARDTHREELAYERNN_H

#include "../NeuralNetwork.h"
#include "../NeuralNetworkLayer.h"

namespace NeuralNetwork {

    namespace FeedforwardNeuralNetwork {

    class FeedforwardThreeLayerNN : public NeuralNetwork
    {
      protected:
        // Layers
        NeuralNetworkLayer *inputLayer;
        NeuralNetworkLayer *hiddenLayer;
        NeuralNetworkLayer *outputLayer;
        // Weights
        double weightsInHid[MAX_NEURONS][MAX_NEURONS];
        double weightsHidOut[MAX_NEURONS][MAX_NEURONS];
        // Properties
        int inputCount, hiddenCount, outputCount;
        int biasIndex;
            /* override */ void initRandomWeights();
            /* override */ void createNetworkLayers();
        void updateWeights(double* inHid, double* hidOut);
      public: 
        FeedforwardThreeLayerNN(int inputCount, int hiddenCount, int outputCount, Utils::Logging::Logger* logger);  
            double* feedForward(double *input);
    };

    }   
}

#endif // FEEDFORWARDTHREELAYERNN_H