ML::RF_params Struct Reference

#include <randomforest.hpp>

Collaboration diagram for ML::RF_params:

Public Attributes
int	n_trees
bool	bootstrap
float	max_samples
uint64_t	seed
int	n_streams
DT::DecisionTreeParams	tree_params

Member Data Documentation

bootstrap

bool ML::RF_params::bootstrap

Control bootstrapping. If bootstrapping is set to true, bootstrapped samples are used for building each tree. Bootstrapped sampling is done by randomly drawing round(max_samples * n_samples) number of samples with replacement. More on bootstrapping: https://en.wikipedia.org/wiki/Bootstrap_aggregating If bootstrapping is set to false, whole dataset is used to build each tree.

max_samples

float ML::RF_params::max_samples

Ratio of dataset rows used while fitting each tree.

n_streams

int ML::RF_params::n_streams

Number of concurrent GPU streams for parallel tree building. Each stream is independently managed by CPU thread. N streams need N times RF workspace.

n_trees

int ML::RF_params::n_trees

Number of decision trees in the random forest.

seed

uint64_t ML::RF_params::seed

Decision tree training hyper parameter struct. random seed

tree_params

DT::DecisionTreeParams ML::RF_params::tree_params

---

The documentation for this struct was generated from the following file:

• include/cuml/ensemble/randomforest.hpp