ranger_regressor: Train and get predictions from a random forest regressor...

In artichaud1/cook: A package for Tidy Hyperparameter Tuning

Description

Train and get predictions from a random forest regressor ('ranger' object).

Usage

ranger_regressor(train_df, target_var, params = list(num.trees = 50), eval_df,
  ...)

Arguments

train_df	Training data.frame.
target_var	Name of the target variable.
params	Parameters passed to the 'ranger' training function.
eval_df	Evaluation data.frame.
...	Additional parameters passed to the 'ranger' training function.

Value

The vector of predictions.

Note

Only binary classification is currently supported.

artichaud1/cook documentation built on May 21, 2019, 9:23 a.m.