Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions | Data Fields
TreeEnsemble Struct Reference

Public Member Functions

 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)
 

Data Fields

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: