LeafBuilder Class#
LeafBuilder #
The base LeafBuilder class from which all other leaf builders must inherit.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
X |
ndarray
|
The feature values used for training. |
required |
Y |
ndarray
|
The response values used for training. |
required |
all_idx |
ndarray
|
A vector specifying samples of the training data that should be considered by the LeafBuilder. |
required |
build_leaf #
Builds a leaf node.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
leaf_id |
int
|
unique identifier of leaf node |
required |
indices |
ndarray
|
indices in leaf node |
required |
depth |
int
|
depth of leaf node |
required |
impurity |
float
|
impurity of leaf node |
required |
weighted_samples |
float
|
summed weight of all samples in the LeafNode |
required |
parent |
DecisionNode
|
parent node |
required |
Returns:
Type | Description |
---|---|
Node
|
built leaf node |