aml_random_forest: AML Random Forest

In jmwerner/ArtisanalMachineLearning: Hand Made Pure Organic Machine Learning

Description

Trains an ensemble of trees via random forest

Usage

aml_random_forest(data, response, b, m = NULL,
  evaluation_criterion = sum_of_squares, min_obs = 5, max_depth = 8,
  verbose = FALSE)

Arguments

data	Input data.frame of dimension n x p for training the random forest
response	Response vector of size nx1 corresponding to the training data
b	Number of bootstrap iterations to perform (trees to build)
m	Number of columns to randomly use at each splitting iteration, defaults to all columns
evaluation_criterion	Function that calculates error criterion for fitting, defaults to sum of squares
min_obs	Minimum observations allowed to end up in a single node, defaults to 5
max_depth	Maximum number of successive splits allowed to happen in the tree, defaults to 8
verbose	Flag to display training updates in the console

Value

Results trained list of class aml_random_forest filled with random forest trees

jmwerner/ArtisanalMachineLearning documentation built on Jan. 7, 2020, 1:50 a.m.