selection.cpp

Go to the documentation of this file.

#include "selection.h"
 
namespace Brush {
namespace Sel {
 
using namespace Brush;
using namespace Pop;
 
template<ProgramType T> 
Selection<T>::Selection()
{
    this->type = "nsga2";
    this->survival = false;
    this->set_operator();
}
 
 
template<ProgramType T> 
Selection<T>::Selection(string type, bool survival)
{
    this->type = type;
    this->survival = survival;
    this->set_operator();
}
 
template<ProgramType T>
void Selection<T>::set_operator()
{
    if (this->type == "nsga2")
        pselector = std::make_shared<NSGA2<T>>(survival);
    else if (this->type == "lexicase")
        pselector = std::make_shared<Lexicase<T>>(survival);
    else
        HANDLE_ERROR_THROW("Undefined Selection Operator " + this->type + "\n");
        
}

template<ProgramType T> 
string Selection<T>::get_type(){ return pselector->name; }

template<ProgramType T> 
void Selection<T>::set_type(string in){ type = in; set_operator();}

template<ProgramType T> 
vector<size_t> Selection<T>::select(Population<T>& pop, int island,  
        const Parameters& params)
{       
    return pselector->select(pop, island, params);
}

template<ProgramType T> 
vector<size_t> Selection<T>::survive(Population<T>& pop, int island, 
        const Parameters& params)
{       
    return pselector->survive(pop, island, params);
}
 
} // Sel
} // Brush
 
template struct Brush::Sel::Selection<Brush::ProgramType::Regressor>;
template struct Brush::Sel::Selection<Brush::ProgramType::BinaryClassifier>;
template struct Brush::Sel::Selection<Brush::ProgramType::MulticlassClassifier>;
template struct Brush::Sel::Selection<Brush::ProgramType::Representer>;