ConnectedNet.h

#ifndef CONNECTED_NET_H
#define CONNECTED_NET_H

#include <vector>

#include "InputNode.h"
#include "Neuron.h"
#include "WeightGenerator.h"
#include "OutputNeuron.h"
#include "TrainingData.h"

class ConnectedNet{
    private:
        std::vector<InputNode*> inputs;
        std::vector<std::vector<Neuron*> > hidden;
        std::vector<OutputNeuron*> outputs;
        InputNode* bias = nullptr;

        std::vector<std::vector<Edge*> > edges;

    public:
        ConnectedNet(int inputNodes, std::vector<int> hiddenLayers,
                int outputNodes, WeightGenerator* weightGen = nullptr);

        ~ConnectedNet();

        std::vector<double> getOutput(std::vector<double> inputValues);

        void train(TrainingData& tData, double learningRate);

        std::vector<double> getDifference();
};

#endif // CONNECTED_NET_H