This learner provides fitting procedures for random forest models, using the randomForest package, using randomForest function.

Format

R6Class object.

Value

Learner object with methods for training and prediction. See Lrnr_base for documentation on learners.

Parameters

ntree = 500

Number of trees to grow. This should not be set to too small a number, to ensure that every input row gets predicted at least a few times.

keep.forest = TRUE

If TRUE, forest is stored, which is required for prediction.

nodesize = 5

Minimum number of observations in terminal nodes.

...

Other parameters passed to randomForest.