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

Description Usage Arguments Value Note

View source: R/ranger_train_predict.R

Description

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

Usage

1
2
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 July 17, 2018, 2:11 a.m.