Brush C++ API
A flexible interpretable machine learning framework
|
Functions | |
template<typename T > | |
ArrayXb | threshold_mask (const T &x, const float &threshold) |
Applies a learned threshold to a feature, returning a mask. | |
template<typename T > requires same_as<typename T::Scalar, bool> | |
ArrayXb | threshold_mask (const T &x, const float &threshold) |
Applies a learned threshold to a feature, returning a mask. | |
template<typename T > requires same_as<typename T::Scalar, bJet> | |
ArrayXb | threshold_mask (const T &x, const float &threshold) |
Applies a learned threshold to a feature, returning a mask. | |
template<typename T > requires same_as<typename T::Scalar, float> | |
ArrayXb | threshold_mask (const T &x, const float &threshold) |
Applies a learned threshold to a feature, returning a mask. | |
template<typename T > requires same_as<typename T::Scalar, fJet> | |
ArrayXb | threshold_mask (const T &x, const float &threshold) |
Applies a learned threshold to a feature, returning a mask. | |
template<typename T > requires same_as<typename T::Scalar, int> | |
ArrayXb | threshold_mask (const T &x, const float &threshold) |
Applies a learned threshold to a feature, returning a mask. | |
template<typename T > requires same_as<typename T::Scalar, iJet> | |
ArrayXb | threshold_mask (const T &x, const float &threshold) |
Applies a learned threshold to a feature, returning a mask. | |
float | gini_impurity_index (const ArrayXf &classes, const vector< float > &uc) |
float | gain (const ArrayXf &lsplit, const ArrayXf &rsplit, bool classification, vector< float > unique_classes) |
template<typename T > | |
vector< float > | get_thresholds (const T &x) |
tuple< string, float > | get_best_variable_and_threshold (const Dataset &d, TreeNode &tn) |
template<typename T > | |
tuple< float, float > | best_threshold (const T &x, const ArrayXf &y, bool classification) |
template<typename T > | |
void | get_best_threshold_by_type (const Dataset &d, auto &results) |
template<typename Ts , std::size_t... Is> | |
auto | get_best_thresholds (const Dataset &d, std::index_sequence< Is... >) |
template<typename T > | |
T | stitch (array< T, 2 > &child_outputs, const ArrayXb &mask) |
Stitches together outputs from left or right child based on threshold. | |
float Split::gain | ( | const ArrayXf & | lsplit, |
const ArrayXf & | rsplit, | ||
bool | classification, | ||
vector< float > | unique_classes ) |