n_fuzzy_fixed_split.h

Go to the documentation of this file.

/* FEAT
copyright 2017 William La Cava
license: GNU/GPL v3
*/
#ifndef NODE_FUZZY_FIXED_SPLIT
#define NODE_FUZZY_FIXED_SPLIT
 
#include "../n_train.h"
 
namespace FT{
 
namespace Pop{
namespace Op{
template <class T>
class NodeFuzzyFixedSplit : public NodeTrain
{
    public:
 
        float threshold; 
        bool threshold_set;
 
        NodeFuzzyFixedSplit();
        
        void set_threshold(ArrayXf& x, VectorXf& y, 
                bool classification);
 
        float gain(const VectorXf& lsplit, const VectorXf& rsplit, 
                bool classification=false, 
                vector<float> unique_classes = vector<float>());
        
        float gini_impurity_index(const VectorXf& classes,
                 vector<float> uc);
        
        void evaluate(const Data& data, State& state);            
 
        void eval_eqn(State& state);
        
    protected:
        NodeFuzzyFixedSplit* clone_impl() const override;
        NodeFuzzyFixedSplit* rnd_clone_impl() const override;
};
// serialization
NLOHMANN_DEFINE_TYPE_NON_INTRUSIVE(NodeFuzzyFixedSplit<float>, name, otype, arity, 
        complexity, visits, train, threshold, threshold_set)
NLOHMANN_DEFINE_TYPE_NON_INTRUSIVE(NodeFuzzyFixedSplit<int>, name, otype, arity, 
        complexity, visits, train, threshold, threshold_set)
} // Op
} // Pop
}// FT  
 
#endif