ensemble/Ensemble.hpp

//
// Created by mateus558 on 29/07/19.
//
 
#pragma once
 
#include <ufjfmltk/core/Learner.hpp>
#include <ufjfmltk/classifier/Classifier.hpp>
 
namespace mltk{
namespace ensemble{        
        template <typename T>
        class Ensemble: virtual public Learner< T > {
            // Attributes
        protected:
            size_t c_size{};
            std::vector< LearnerPointer< T > > m_learners;
            Solution solution;
        public:
            Ensemble() = default;
            explicit Ensemble(DataPointer< T > samples): Learner< T > (samples) {}
            /*********************************************
             *               Getters                     *
             *********************************************/
 
            [[nodiscard]] Solution getSolution() const {return solution;}
            Solution *getSolutionRef() { return &solution; }
 
            /*********************************************
             *               Setters                     *
             *********************************************/
 
            void setLearner(Learner<T> *learner) {
                Ensemble::learner = learner;
            }
 
            void setSamples(DataPointer<T> samples) override {
                Ensemble::samples = samples;
                for (auto &m_learner : m_learners) {
                    m_learner->setSamples(samples);
                }
            }
 
            void setCommitteeSize(size_t c_size) {
                Ensemble::c_size = c_size;
            }
 
            size_t size(){
                return this->m_learners.size();
            }
 
            auto begin(){
                return this->m_learners.begin();
            }
 
            auto end(){
                return this->m_learners.end();
            }
 
            LearnerPointer<T> operator[](size_t idx) const  {
                return this->m_learners[idx];
            }
 
            auto learners(){
                return this->m_learners;
            }
 
        };
}
}