tournamentSelector.h

 
#ifndef TOURNAMENT_SELECTOR_H
#define TOURNAMENT_SELECTOR_H
 
#include "selector/selector.h"
 
namespace Selector {
 
    class TournamentSelector : public Selector
    {
      protected:
        std::set<const TPG::TPGVertex*> verticesToDelete;
 
      public:
        TournamentSelector(std::shared_ptr<TPG::TPGGraph> graph,
                           const Learn::LearningParameters& params)
            : Selector{graph, params}
        {
        }
 
        virtual void launchSelection(
            std::multimap<std::shared_ptr<Learn::EvaluationResult>,
                          const TPG::TPGVertex*>& results,
            Mutator::RNG& rng) override;
 
        virtual void doSelection(
            std::multimap<std::shared_ptr<Learn::EvaluationResult>,
                          const TPG::TPGVertex*>& results,
            Mutator::RNG& rng) override;
 
        void addToVerticesToDelete(const TPG::TPGVertex* vertex);
 
        virtual const SelectionContext& updateContext() override;
 
        virtual void updateAfterPopulate() override;
 
        virtual const std::set<const TPG::TPGVertex*>& getVerticesToDelete();
    };
}; // namespace Selector
 
#endif // TOURNAMENT_SELECTOR_H