circularRegressor.hpp

Go to the documentation of this file.

#ifndef CIRCULARREGRESSOR_HPP
#define CIRCULARREGRESSOR_HPP

#include <canopy/circularRegressor/vonMisesDistribution.hpp>
#include <canopy/randomForestBase/randomForestBase.hpp>

namespace canopy
{

template <unsigned TNumParams>
class circularRegressor : public randomForestBase<circularRegressor<TNumParams>,float,vonMisesDistribution,vonMisesDistribution,TNumParams>
{
    public:

        // Methods
        // -------
        circularRegressor();
        circularRegressor(const int num_trees, const int num_levels, const float info_gain_tresh = C_DEFAULT_MIN_INFO_GAIN);

    protected:
        typedef typename randomForestBase<circularRegressor<TNumParams>,float,vonMisesDistribution,vonMisesDistribution,TNumParams>::scoreInternalIndexStruct scoreInternalIndexStruct;

        // Methods
        // -------

        void initialiseNodeDist(const int t, const int n);
        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();
        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;
        float minInfoGain(const int /*tree*/, const int /*node*/) const;
        void printHeaderDescription(std::ofstream& /*stream*/) const;
        void printHeaderData(std::ofstream& /*stream*/) const;
        void readHeader(std::ifstream& /*stream*/);

        // Data
        // ----
        std::vector<double> sin_precalc; 
        std::vector<double> cos_precalc; 
        float min_info_gain; 

        // Constants
        // ---------
        static constexpr int C_NUM_SPLIT_TRIALS = 100; 
        static constexpr float C_DEFAULT_MIN_INFO_GAIN = 0.1; 
};

} // end of namespace

#include <canopy/circularRegressor/circularRegressor.tpp>
#endif
// CIRCULARREGRESSOR_HPP