classifier.hpp

Go to the documentation of this file.

#ifndef CLASSIFER_HPP
#define CLASSIFER_HPP

#include <canopy/randomForestBase/randomForestBase.hpp>
#include <canopy/classifier/discreteDistribution.hpp>

namespace canopy
{

template <unsigned TNumParams>
class classifier : public randomForestBase<classifier<TNumParams>,int,discreteDistribution,discreteDistribution,TNumParams>
{
    public:
        // Methods
        classifier(const int num_classes, const int num_trees, const int num_levels, const double info_gain_tresh = C_DEFAULT_MIN_INFO_GAIN); // constructor
        classifier();
        int getNumberClasses() const;
        void setClassNames(const std::vector<std::string>& new_class_names);
        void getClassNames(std::vector<std::string>& end_class_names) const;
        void raiseNodeTemperature(const double T);

    protected:
        typedef typename randomForestBase<classifier<TNumParams>,int,discreteDistribution,discreteDistribution,TNumParams>::scoreInternalIndexStruct scoreInternalIndexStruct;

        // Methods
        void initialiseNodeDist(const int t, const int n);
        template <class TLabelIterator>
        void bestSplit(const std::vector<scoreInternalIndexStruct> &data_structs, const TLabelIterator first_label, const int /*tree*/, const int /*node*/, const float initial_impurity,float& info_gain, float& thresh) const;
        void printHeaderDescription(std::ofstream &stream) const;
        void printHeaderData(std::ofstream &stream) const;
        void readHeader(std::ifstream &stream);
        float minInfoGain(const int /*tree*/, const int /*node*/) const;
        template <class TLabelIterator>
        float singleNodeImpurity(const TLabelIterator first_label, const std::vector<int>& nodebag, const int /*tree*/, const int /*node*/) const;
        template <class TLabelIterator, class TIdIterator>
        void trainingPrecalculations(const TLabelIterator first_label, const TLabelIterator last_label, const TIdIterator/*unused*/);
        void cleanupPrecalculations();

        // Data
        int n_classes; 
        std::vector<std::string> class_names; 
        std::vector<double> xlogx_precalc; 
        double min_info_gain; 

        // Constants
        static constexpr double C_DEFAULT_MIN_INFO_GAIN = 0.05; 
};

} // end of namespace

#include <canopy/classifier/classifier.tpp>
#endif
// CLASSIFER_HPP