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
 

Functions

tuple< string, floatBrush::Split::get_best_variable_and_threshold (const Dataset &d, TreeNode &tn)
 
template<>
vector< floatBrush::Split::get_thresholds< ArrayXb > (const ArrayXb &x)
 
template<>
vector< floatBrush::Split::get_thresholds< ArrayXbJet > (const ArrayXbJet &x)
 
template<>
vector< floatBrush::Split::get_thresholds< ArrayXi > (const ArrayXi &x)
 
template<>
vector< floatBrush::Split::get_thresholds< ArrayXiJet > (const ArrayXiJet &x)
 
template<>
vector< floatBrush::Split::get_thresholds< ArrayXf > (const ArrayXf &x)
 
template<>
vector< floatBrush::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 > &unique_classes)