Brush C++ API
A flexible interpretable machine learning framework
Loading...
Searching...
No Matches
split.cpp File Reference
#include "operator.h"
#include <utility>
Include dependency graph for split.cpp:

Go to the source code of this file.

Namespaces

namespace  Brush
 < nsga2 selection operator for getting the front
 
namespace  Brush::Split
 

Typedefs

typedef std::variant< ArrayXb, ArrayXi, ArrayXf, ArrayXXb, ArrayXXi, ArrayXXf, TimeSeriesb, TimeSeriesi, TimeSeriesf, ArrayXbJet, ArrayXiJet, ArrayXfJet, ArrayXXbJet, ArrayXXiJet, ArrayXXfJet, TimeSeriesbJet, TimeSeriesiJet, TimeSeriesfJet > State
 defines the possible types of data flowing thru nodes.
 

Functions

tuple< string, float > Brush::Split::get_best_variable_and_threshold (const Dataset &d, TreeNode &tn)
 
template<>
vector< float > Brush::Split::get_thresholds< ArrayXb > (const ArrayXb &x)
 
template<>
vector< float > Brush::Split::get_thresholds< ArrayXbJet > (const ArrayXbJet &x)
 
template<>
vector< float > Brush::Split::get_thresholds< ArrayXi > (const ArrayXi &x)
 
template<>
vector< float > Brush::Split::get_thresholds< ArrayXiJet > (const ArrayXiJet &x)
 
template<>
vector< float > Brush::Split::get_thresholds< ArrayXf > (const ArrayXf &x)
 
template<>
vector< float > Brush::Split::get_thresholds< ArrayXfJet > (const ArrayXfJet &x)
 
template<>
ArrayXb Brush::Split::threshold_mask< State > (const State &x, const float &threshold)
 
float Brush::Split::gain (const ArrayXf &lsplit, const ArrayXf &rsplit, bool classification, vector< float > unique_classes)
 
float Brush::Split::gini_impurity_index (const ArrayXf &classes, const vector< float > &uc)
 

Typedef Documentation

◆ State

typedef std::variant< ArrayXb, ArrayXi, ArrayXf, ArrayXXb, ArrayXXi, ArrayXXf, TimeSeriesb, TimeSeriesi, TimeSeriesf, ArrayXbJet, ArrayXiJet, ArrayXfJet, ArrayXXbJet, ArrayXXiJet, ArrayXXfJet, TimeSeriesbJet, TimeSeriesiJet, TimeSeriesfJet > Brush::Data::State

defines the possible types of data flowing thru nodes.

Definition at line 140 of file types.h.