featselect/FeatureSelection.hpp

//
// Created by Mateus Coutinho Marim on 6/26/2018.
//
 
#pragma once
 
#include "ufjfmltk/core/Data.hpp"
#include "ufjfmltk/valid/Validation.hpp"
#include <memory>
 
namespace mltk{
    namespace featselect {
        template<typename T>
        class FeatureSelection {
        protected:
            std::shared_ptr<Data<T> > samples;
            classifier::Classifier<T> *classifier;
            validation::CrossValidation *cv;
            Solution solution;
            Timer timer;
 
            const int MAX_DEPTH = 99999;
            const int MAX_BREATH = 99999;
            const double NUM_ERROR_EPS = 0.05;
            int depth = 0;
            int final_dim = 0;
            int jump = 0;
            int skip = 0;
            bool leave_one_out = false;
            int verbose = 0;
 
        public:
 
            virtual Data<T> selectFeatures() = 0;
 
            /*********************************************
             *               Setters                     *
             *********************************************/
 
            void setSamples(const std::shared_ptr<Data<T> > &samples) { this->samples = samples; }
            void setSamples(const Data<T>& samples) { this->samples = mltk::make_data<T>(samples); }
            void setClassifier(classifier::Classifier<T> *classifier) {
                this->classifier = classifier;
                this->classifier->setVerbose(0);
            }
 
            void setFinalDimension(int final_dim) { this->final_dim = final_dim; }
 
            void setDepth(int depth) { this->depth = depth; }
 
            void setJump(int jump) { this->jump = jump; }
 
            void setSkip(int skip) { this->skip = skip; }
 
            void setVerbose(int verbose) { this->verbose = verbose; }
 
            void setCrossValidation(validation::CrossValidation *cv) { this->cv = cv; }
        };
    }
}