selection.h

Go to the documentation of this file.

/* Brush
copyright 2020 William La Cava
license: GNU/GPL v3
*/
 
#ifndef SELECTION_H
#define SELECTION_H
 
#include "selection_operator.h"
#include "nsga2.h"
#include "lexicase.h"
 
namespace Brush {
namespace Sel {
 
using namespace Brush;
using namespace Pop;

template<ProgramType T>
struct Selection
{
public:
    std::shared_ptr<SelectionOperator<T>> pselector;
    string type;
    bool survival;
    
    Selection();
    ~Selection(){};
    Selection(string type, bool survival);
 
    void set_operator();
    
    string get_type();
    void set_type(string);
    
    vector<size_t> select(Population<T>& pop, int island, 
            const Parameters& params);
    
    vector<size_t> survive(Population<T>& pop, int island,  
            const Parameters& params);
};
 
extern template struct Selection<PT::Regressor>;
extern template struct Selection<PT::BinaryClassifier>;
extern template struct Selection<PT::MulticlassClassifier>;
extern template struct Selection<PT::Representer>;
 
} // Sel
} // Brush
#endif