NetTraining.h

#ifndef NET_TRAINING_H
#define NET_TRAINING_H

#include <vector>

#include "ConnectedNet.h"
#include "StopCondition.h"

class NetTraining{
    private:
        ConnectedNet* net = nullptr;
        std::vector<TrainingData> trainingData;
        std::vector<TrainingData> evalData;
        StopCondition* stopper = nullptr;
        double learningRate = 1.0;
        int trainingPerIteration = 1;

        double error = 0;
        int iteration = 0;

        void printInfo();

        static void printInOut(const std::vector<double>& input,
                const std::vector<double>& output);

    public:
        NetTraining(ConnectedNet* net, std::vector<TrainingData> trainingData,
                std::vector<TrainingData> evalData, StopCondition* stopper,
                double learningRate, int trainingPerIteration = 1);

        void run();
};

#endif // NET_TRAINING_H