selection_operator.h

Go to the documentation of this file.

#ifndef SELECTION_OPERATOR_H
#define SELECTION_OPERATOR_H
 
// virtual class. selection must be made with static methods
 
// #include "../init.h"
// #include "../data/data.h"
// #include "../types.h"
// #include "../params.h"
#include "../pop/population.h"
 
namespace Brush {
namespace Sel {
 
using namespace Brush;
using namespace Pop;
 
// TODO: stop having to give the template for the selection operator (and also other places like pop)
 
template<ProgramType T> 
class SelectionOperator
{
public:
    bool survival; 
    string name; 

    virtual ~SelectionOperator();
        
    virtual vector<size_t> select(Population<T>& pop, int island, const Parameters& p);
    
    virtual vector<size_t> survive(Population<T>& pop, int island, const Parameters& p);
};
 
extern template class SelectionOperator<PT::Regressor>;
extern template class SelectionOperator<PT::BinaryClassifier>;
extern template class SelectionOperator<PT::MulticlassClassifier>;
extern template class SelectionOperator<PT::Representer>;
 
} // selection
} // Brush
#endif