Neuron.h

#ifndef NEURON_H
#define NEURON_H

#include <functional>
#include <vector>
#include "Node.h"
#include "Edge.h"

// Forward declaration
class Edge;

class Neuron: public Node {
    private:
        std::vector<Edge*> inputs;
        std::vector<Edge*> outputs;

        double output = 0;

        std::function<double(double)> activationFunc;

    protected:
        double delta = 0;

        double sigmoidDelta();

    public:
        Neuron(std::function<double(double)> activationFunc,
                std::vector<Edge*> inputs,
                std::vector<Edge*> outputs);

        Neuron(std::function<double(double)> activationFunc);

        void setInputs(std::vector<Edge*> inputs);

        void setOutputs(std::vector<Edge*> outputs);

        void calcOutput();

        void calcDelta();

        double getOutput();

        double getDelta();
};

#endif // NEURON_H