Brush C++ API
A flexible interpretable machine learning framework
|
Functions | |
tuple< string, float > | get_best_variable_and_threshold (const Dataset &d, TreeNode &tn) |
template<> | |
vector< float > | get_thresholds< ArrayXb > (const ArrayXb &x) |
template<> | |
vector< float > | get_thresholds< ArrayXbJet > (const ArrayXbJet &x) |
template<> | |
vector< float > | get_thresholds< ArrayXi > (const ArrayXi &x) |
template<> | |
vector< float > | get_thresholds< ArrayXiJet > (const ArrayXiJet &x) |
template<> | |
vector< float > | get_thresholds< ArrayXf > (const ArrayXf &x) |
template<> | |
vector< float > | get_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) |
vector< float > Brush::Split::get_thresholds< ArrayXbJet > | ( | const ArrayXbJet & | x | ) |
vector< float > Brush::Split::get_thresholds< ArrayXfJet > | ( | const ArrayXfJet & | x | ) |
vector< float > Brush::Split::get_thresholds< ArrayXiJet > | ( | const ArrayXiJet & | x | ) |