classificationSelectionMetrics.h

 
#ifndef CLASSIFICATION_SELECTION_METRICS_H
#define CLASSIFICATION_SELECTION_METRICS_H
 
#include <numeric>
 
#include "selector/selectionMetrics.h"
 
namespace Selector {
    class ClassificationSelectionMetrics : public SelectionMetrics
    {
      protected:
        std::vector<double> scorePerClass;
 
        std::vector<size_t> nbEvalPerClass;
 
      public:
        ClassificationSelectionMetrics() = default;
 
        ClassificationSelectionMetrics(
            const std::vector<double>& scorePerClass,
            const std::vector<size_t>& nbEvalPerClass)
            : SelectionMetrics(std::accumulate(scorePerClass.cbegin(),
                                               scorePerClass.cend(), 0.0) /
                               scorePerClass.size()),
              scorePerClass{scorePerClass}, nbEvalPerClass{nbEvalPerClass}
        {
            if (scorePerClass.size() != nbEvalPerClass.size()) {
                throw std::runtime_error("Number of class missmatch.");
            }
        };
 
        virtual const std::vector<double>& getScorePerClass() const;
 
        virtual const std::vector<size_t>& getNbEvalPerClassPerClass() const;
 
        void initMetrics(
            const TPG::TPGVertex* agent,
            const Learn::LearningEnvironment& learningEnvironment) override;
 
        void extractMetricsEpisode(
            const TPG::TPGVertex* agent, size_t nbStepsExecuted,
            const Learn::LearningEnvironment& learningEnvironment) override;
 
        virtual void weightedSum(std::shared_ptr<SelectionMetrics> other,
                                 size_t nbEvaluation,
                                 size_t nbEvaluationOther) override;
    };
 
}; // namespace Selector
 
#endif // CLASSIFICATION_SELECTION_METRICS_H