|
| TreeEnsemble (int *children_left, int *children_right, int *children_default, int *features, tfloat *thresholds, tfloat *values, tfloat *node_sample_weights, unsigned max_depth, unsigned tree_limit, tfloat base_offset, unsigned max_nodes, unsigned num_outputs) |
|
void | get_tree (TreeEnsemble &tree, const unsigned i) const |
|
void | allocate (unsigned tree_limit_in, unsigned max_nodes_in, unsigned num_outputs_in) |
|
void | free () |
|
void | fill_adjusted_tree (int node_index, ExplanationDataset &instance, const int *mask, unsigned *feature_sets, TreeEnsemble &adjusted) |
|
void | create_adjusted_tree (ExplanationDataset &instance, const int *mask, unsigned *feature_sets, TreeEnsemble &adjusted) |
|
void | calc_feature_contribs_conditional (MedMat< float > &mat_x_in, unordered_map< string, float > contiditional_variables, MedMat< float > &mat_x_out, MedMat< float > &mat_contribs) |
|
tfloat | predict (ExplanationDataset &instance, int node_index) |
|
|
int * | children_left |
|
int * | children_right |
|
int * | children_default |
|
int * | features |
|
tfloat * | thresholds |
|
tfloat * | values |
|
tfloat * | node_sample_weights |
|
unsigned | max_depth |
|
unsigned | tree_limit |
|
tfloat | base_offset |
|
unsigned | max_nodes |
|
unsigned | num_outputs |
|
bool | is_allocate |
|
The documentation for this struct was generated from the following files:
- Internal/MedAlgo/MedAlgo/tree_shap.h
- Internal/MedAlgo/MedAlgo/tree_shap.cpp