classificationSelector.h

 
#ifndef CLASSIFICATION_SELECTOR_H
#define CLASSIFICATION_SELECTOR_H
 
#include "selector/selector.h"
 
namespace Selector {
 
    class ClassificationSelector : public Selector
    {
      protected:
        uint64_t nbActions;
 
      public:
        ClassificationSelector(std::shared_ptr<TPG::TPGGraph> graph,
                               const Learn::LearningParameters& params,
                               uint64_t nbActions)
            : Selector{graph, params}, nbActions{nbActions}
        {
        }
 
        virtual std::shared_ptr<SelectionMetrics> createSelectionMetrics()
            override;
 
        virtual void doSelection(
            std::multimap<std::shared_ptr<Learn::EvaluationResult>,
                          const TPG::TPGVertex*>& results,
            Mutator::RNG& rng) override;
    };
}; // namespace Selector
 
#endif // CLASSIFICATION_SELECTOR_H