classifier/Classifier.hpp

Go to the documentation of this file.

 
#ifndef CLASSIFIER__HPP
#define CLASSIFIER__HPP
 
 
#include "ufjfmltk/core/Learner.hpp"
namespace mltk{
    namespace classifier {
        template<typename T>
        class Classifier : virtual public Learner<T> {
            // Attributes
        protected :
            bool hasInitialSolution = false;
            std::vector<Point<T> > svs;
            Solution solution;
            double gamma = 0;
 
            // Operations
        public :
 
            Classifier<T>() {}
 
            Classifier<T>(DataPointer<T> samples) : Learner<T>(samples) {}
 
            Classifier<T>(const Classifier<T> &classifier)
                    : Learner<T>(classifier) {
                this->hasInitialSolution = classifier.hasInitialSolution;
                this->gamma = classifier.gamma;
            }
 
            virtual double evaluateProbability(const mltk::Point<double>& p){return 0.0;}
            mltk::Point<int> batchEvaluateProbability(const mltk::Data<T>& data);
            /*********************************************
             *               Getters                     *
             *********************************************/
 
            Solution getSolution() const { return solution; }
 
            Solution *getSolutionRef() { return &solution; }
 
            /*********************************************
             *               Setters                     *
             *********************************************/
            inline void setGamma(double gamma) { this->gamma = gamma; }
 
            void setW(std::vector<double> w) { this->solution.w = w; }
 
            void setSolution(Solution solution) { this->solution = solution; }
        };
 
        template<typename T>
        mltk::Point<int> Classifier<T>::batchEvaluateProbability(const Data <T> &data) {
            mltk::Point<int> preds(data.size());
            std::transform(data.begin(), data.end(), preds.begin(), [this](auto point){
                return this->evaluateProbability(*point);
            });
            return preds;
        }
    }
}
 
#endif