timingSelectionMetrics.h

#ifndef TIMING_SELECTION_METRICS_H
#define TIMING_SELECTION_METRICS_H
 
#include "selector/selectionMetrics.h"
 
namespace Selector {
    class TimingSelectionMetrics : public SelectionMetrics
    {
      protected:
        std::shared_ptr<SelectionMetrics> wrapped;
 
        double agentTime = 0;
 
        double leTime = 0;
 
        size_t nbActions = 0;
 
      public:
        virtual ~TimingSelectionMetrics() = default;
 
        double getAgentTime() const
        {
            return agentTime;
        }
 
        double getLeTime() const
        {
            return leTime;
        }
 
        size_t getNbActions() const
        {
            return nbActions;
        }
 
        double getScore() const override
        {
            return wrapped->getScore();
        };
 
        double getUtility() const override
        {
            return wrapped->getUtility();
        };
 
        explicit TimingSelectionMetrics(std::shared_ptr<SelectionMetrics> obj)
            : wrapped(obj){};
 
        void extractMetricsStep(
            const TPG::TPGVertex* agent, const std::vector<double> actionValues,
            const Learn::LearningEnvironment& learningEnvironment) override;
 
        void extractMetricsEpisodeWithTiming(
            const TPG::TPGVertex* agent, size_t nbStepsExecuted,
            const Learn::LearningEnvironment& learningEnvironment,
            double agentTimeEpisode, double leTimeEpisode);
 
        void weightedSum(std::shared_ptr<SelectionMetrics> other,
                         size_t nbEvaluation,
                         size_t nbEvaluationOther) override;
    };
 
    bool operator<(std::shared_ptr<TimingSelectionMetrics> a,
                   std::shared_ptr<TimingSelectionMetrics> b);
 
    bool operator<(std::shared_ptr<SelectionMetrics> a,
                   std::shared_ptr<TimingSelectionMetrics> b);
 
    bool operator<(std::shared_ptr<TimingSelectionMetrics> a,
                   std::shared_ptr<SelectionMetrics> b);
}; // namespace Selector
 
#endif // TIMING_SELECTION_METRICS_H