Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!
Get Started for Free

tune_forest

Tune a forests


Description

Finds the optimal parameters to be used in training a forest.

Usage

tune_forest(
  data,
  nrow.X,
  ncol.X,
  args,
  tune.parameters,
  tune.parameters.defaults,
  num.fit.trees,
  num.fit.reps,
  num.optimize.reps,
  train
)

Arguments

data

The data arguments (output from create_train_matrices) for the forest.

nrow.X

The number of observations.

ncol.X

The number of variables.

args

The remaining call arguments for the forest.

tune.parameters

The vector of parameter names to tune.

tune.parameters.defaults

The grf default values for the vector of parameter names to tune.

num.fit.trees

The number of trees in each 'mini forest' used to fit the tuning model.

num.fit.reps

The number of forests used to fit the tuning model.

num.optimize.reps

The number of random parameter values considered when using the model to select the optimal parameters.

train

The grf forest training function.

Value

tuning output


grf

Generalized Random Forests

v1.2.0
GPL-3
Authors
Julie Tibshirani [aut, cre], Susan Athey [aut], Rina Friedberg [ctb], Vitor Hadad [ctb], David Hirshberg [ctb], Luke Miner [ctb], Erik Sverdrup [ctb], Stefan Wager [aut], Marvin Wright [ctb]
Initial release

We don't support your browser anymore

Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.