classificationLearningEnvironment.h

 
#ifndef CLASSIFICATION_LEARNING_ENVIRONMENT_H
#define CLASSIFICATION_LEARNING_ENVIRONMENT_H
 
#include <vector>
 
#include "learn/learningEnvironment.h"
 
namespace Learn {
    class ClassificationLearningEnvironment : public LearningEnvironment
    {
      protected:
        std::vector<std::vector<uint64_t>> classificationTable;
 
        uint64_t currentClass;
 
      public:
        ClassificationLearningEnvironment(uint64_t nbClass)
            : LearningEnvironment(nbClass), currentClass{0},
              classificationTable(nbClass, std::vector<uint64_t>(nbClass, 0)){};
 
        const std::vector<std::vector<uint64_t>>& getClassificationTable()
            const;
 
        virtual void doAction(double actionID) override = 0;
 
        virtual double getScore() const override;
 
        virtual void reset(size_t seed = 0,
                           LearningMode mode = LearningMode::TRAINING,
                           uint16_t iterationNumber = 0,
                           uint64_t generationNumber = 0) override = 0;
    };
}; // namespace Learn
 
#endif