PrimalClassifier.hpp

Go to the documentation of this file.

 
#ifndef PRIMALCLASSIFIER__HPP
#define PRIMALCLASSIFIER__HPP
 
#include "Classifier.hpp"
 
namespace mltk{
        namespace classifier {
            template<typename T>
            class PrimalClassifier : virtual public Classifier<T> {
                // Attributes
            protected :
                std::vector<double> w;
                double q = 2, p = 2;
 
            protected:
                double flexible = 0.0;
                double alpha_aprox = 0.0;
            public:
 
                PrimalClassifier<T>() {}
                PrimalClassifier<T>(DataPointer<T> samples) : Classifier<T>(samples) {}
                PrimalClassifier<T>(mltk::Data<T>& samples) {
                    this->samples = mltk::make_data<T>(samples);
                }
                PrimalClassifier<T>(const PrimalClassifier<T> &primal_learner)
                        : Classifier<T>(primal_learner) {
                    this->q = primal_learner.q;
                    this->p = primal_learner.p;
                    this->flexible = primal_learner.flexible;
                    this->alpha_aprox = primal_learner.alpha_aprox;
                }
 
                double evaluate(const Point<T> &p, bool raw_value = false) override {
                    double func = 0.0;
                    size_t i, dim = this->solution.w.size();
 
                    if (p.X().size() != dim) {
                        std::cerr << "The point must have the same dimension of the feature set! (" << p.X().size()
                                  << ", " << dim << ")" << std::endl;
                        return 0;
                    }
 
                    for (func = this->solution.bias, i = 0; i < dim; i++) {
                        func += this->solution.w[i] * p[i];
                    }
 
                    return (func >= 0) ? 1 : -1;
                }
 
                /*********************************************
                 *               Getters                     *
                 *********************************************/
 
                std::string getFormulationString() override { return "Primal"; }
 
                double getP() const { return p; }
 
                double getQ() const { return q; }
 
                /*********************************************
                 *               Setters                     *
                 *********************************************/
 
                void setqNorm(double q) { this->q = q; }
 
                void setpNorm(double p) { this->p = p; }
 
                void setFlexible(double flexible) { this->flexible = flexible; }
 
                void setAlphaAprox(double alpha_aprox) { this->alpha_aprox = alpha_aprox; }
            };
        }
}
#endif