bayeso.trees.trees_random_forest

It defines a random forest.

bayeso.trees.trees_random_forest.get_random_forest(X: ndarray, Y: 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