Brush C++ API
A flexible interpretable machine learning framework
Loading...
Searching...
No Matches
Brush::Split Namespace Reference

Functions

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

Function Documentation

◆ gain()

float Brush::Split::gain ( const ArrayXf & lsplit,
const ArrayXf & rsplit,
bool classification,
vector< float > unique_classes )

Definition at line 66 of file split.cpp.

Here is the call graph for this function:

◆ get_best_variable_and_threshold()

tuple< string, float > Brush::Split::get_best_variable_and_threshold ( const Dataset & d,
TreeNode & tn )

Definition at line 7 of file split.cpp.

Here is the call graph for this function:

◆ get_thresholds< ArrayXb >()

template<>
vector< float > Brush::Split::get_thresholds< ArrayXb > ( const ArrayXb & x)

Definition at line 7 of file split.cpp.

◆ get_thresholds< ArrayXbJet >()

template<>
vector< float > Brush::Split::get_thresholds< ArrayXbJet > ( const ArrayXbJet & x)

Definition at line 7 of file split.cpp.

◆ get_thresholds< ArrayXf >()

template<>
vector< float > Brush::Split::get_thresholds< ArrayXf > ( const ArrayXf & x)

Definition at line 7 of file split.cpp.

◆ get_thresholds< ArrayXfJet >()

template<>
vector< float > Brush::Split::get_thresholds< ArrayXfJet > ( const ArrayXfJet & x)

Definition at line 7 of file split.cpp.

◆ get_thresholds< ArrayXi >()

template<>
vector< float > Brush::Split::get_thresholds< ArrayXi > ( const ArrayXi & x)

Definition at line 7 of file split.cpp.

◆ get_thresholds< ArrayXiJet >()

template<>
vector< float > Brush::Split::get_thresholds< ArrayXiJet > ( const ArrayXiJet & x)

Definition at line 7 of file split.cpp.

◆ gini_impurity_index()

float Brush::Split::gini_impurity_index ( const ArrayXf & classes,
const vector< float > & unique_classes )

Definition at line 93 of file split.cpp.

Here is the call graph for this function:
Here is the caller graph for this function:

◆ threshold_mask< State >()

template<>
ArrayXb Brush::Split::threshold_mask< State > ( const State & x,
const float & threshold )

Definition at line 54 of file split.cpp.

Here is the call graph for this function: