36#ifndef CLASSIFICATION_LEARNING_ENVIRONMENT_H
37#define CLASSIFICATION_LEARNING_ENVIRONMENT_H
41#include "learn/learningEnvironment.h"
95 virtual void doAction(uint64_t actionID)
override = 0;
105 virtual double getScore()
const override;
114 LearningMode mode = LearningMode::TRAINING)
override = 0;
Specialization of the LearningEnvironment class for classification purposes.
Definition: classificationLearningEnvironment.h:49
virtual void doAction(uint64_t actionID) override=0
Default implementation for the doAction method.
Definition: classificationLearningEnvironment.cpp:41
const std::vector< std::vector< uint64_t > > & getClassificationTable() const
Get a const ref to the classification table of the learning environment.
Definition: classificationLearningEnvironment.cpp:51
virtual double getScore() const override
Default scoring for classificationLearningEnvironment.
Definition: classificationLearningEnvironment.cpp:56
std::vector< std::vector< uint64_t > > classificationTable
2D array storing for each class the guesses that were made by the LearningAgent.
Definition: classificationLearningEnvironment.h:59
uint64_t currentClass
Class of the current data.
Definition: classificationLearningEnvironment.h:67
ClassificationLearningEnvironment(uint64_t nbClass)
Definition: classificationLearningEnvironment.h:77
virtual void reset(size_t seed=0, LearningMode mode=LearningMode::TRAINING) override=0
Default implementation of the reset.
Definition: classificationLearningEnvironment.cpp:97
Interface for creating a Learning Environment.
Definition: learningEnvironment.h:80
Definition: adversarialEvaluationResult.h:45
LearningMode
Different modes in which the LearningEnvironment can be reset.
Definition: learningEnvironment.h:58