mapElitesSelectionMetrics.h

 
 
#ifndef MAP_ELITES_SELECTION_METRICS_H
#define MAP_ELITES_SELECTION_METRICS_H
 
#include <numeric>
 
#include "selector/mapElites/mapElitesDescriptor.h"
#include "selector/selectionMetrics.h"
 
namespace Selector {
 
    namespace MapElites {
 
        class MapElitesSelectionMetrics : public SelectionMetrics
        {
          protected:
            std::map<std::shared_ptr<const MapElitesDescriptor>,
                     std::vector<double>>
                mapDescriptors;
 
          public:
            MapElitesSelectionMetrics(
                std::vector<std::shared_ptr<const MapElitesDescriptor>>
                    descriptors)
            {
                for (auto descriptor : descriptors) {
                    mapDescriptors.insert({descriptor, {}});
                }
            }
 
            MapElitesSelectionMetrics(
                double score,
                const std::map<std::shared_ptr<const MapElitesDescriptor>,
                               std::vector<double>>& mapDescriptors)
                : SelectionMetrics(score), mapDescriptors(mapDescriptors){};
 
            MapElitesSelectionMetrics(double score) : SelectionMetrics(score){};
 
            const std::map<std::shared_ptr<const MapElitesDescriptor>,
                           std::vector<double>>&
            getMapDescriptors() const;
 
            void initMetrics(
                const TPG::TPGVertex* agent,
                const Learn::LearningEnvironment& learningEnvironment) override;
 
            void extractMetricsStep(
                const TPG::TPGVertex* agent, std::vector<double> actionValues,
                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 MapElites
 
}; // namespace Selector
 
#endif // MAP_ELITES_SELECTION_METRICS_H