regressor/Regressor.hpp

//
// Created by mateus558 on 18/10/18.
//
 
#pragma once
 
#include "ufjfmltk/core/Learner.hpp"
 
namespace mltk{
        namespace regressor {
            
            template<typename T>
            class Regressor : public Learner<T> {
            protected:
                Solution solution;
            public:
                Regressor() = default;
 
                explicit Regressor(DataPointer <T> samples) : Learner<T>(samples) {}
 
                /*********************************************
                 *               Getters                     *
                 *********************************************/
 
                [[nodiscard]] Solution getSolution() const { return solution; }
 
                Solution *getSolutionRef() { return &solution; }
 
                /*********************************************
                 *               Setters                     *
                 *********************************************/
 
                void setW(const std::vector<double> &w) { this->solution.w = w; }
 
                void setSolution(Solution solution) { this->solution = solution; }
            };
        }
}