bayeso.trees.trees_random_forest¶
It defines a random forest.
-
bayeso.trees.trees_random_forest.
get_random_forest
(X: numpy.ndarray, Y: numpy.ndarray, num_trees: int, depth_max: int, size_min_leaf: int, num_features: int) → list¶ It returns a random forest.
Parameters: - X (np.ndarray) – inputs. Shape: (N, d).
- Y (str.) – outputs. Shape: (N, 1).
- num_trees (int.) – the number of trees.
- depth_max (int.) – maximum depth of tree.
- size_min_leaf (int.) – minimum size of leaf.
- num_features (int.) – the number of split features.
Returns: list of trees
Return type: list
Raises: AssertionError