n_split.h

Go to the documentation of this file.

/* FEAT
copyright 2017 William La Cava
license: GNU/GPL v3
*/
#ifndef NODE_SPLIT
#define NODE_SPLIT
 
#include "../n_train.h"
 
namespace FT{
namespace Pop{
namespace Op{
 
template <class T>
class NodeSplit : public NodeTrain
{
    public:
 
        float threshold; 
 
        NodeSplit();
        
        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:
        NodeSplit* clone_impl() const override;
        NodeSplit* rnd_clone_impl() const override;
};
// serialization
NLOHMANN_DEFINE_TYPE_NON_INTRUSIVE(NodeSplit<float>, name, otype, arity, complexity, 
        visits, train, threshold)
NLOHMANN_DEFINE_TYPE_NON_INTRUSIVE(NodeSplit<int>, name, otype, arity, complexity, 
        visits, train, threshold)
}
}
}   
 
#endif