train_rf: Train a Random Forest model

Description Usage Arguments Value

View source: R/classification.R

Description

Trains a Random Forest based on a training data set containing computed features

Usage

1
2
train_rf(train, formula = response ~ . - id, importance = FALSE,
  ntree = 500, mtry = floor(sqrt(ncol(train) - 2)))

Arguments

train

A training data set as produced by make_train

formula

The formula to use

importance

Boolean if the variable importance should be measured (default FALSE)

ntree

Number of trees to grow

mtry

Number of variables randomly sampled as candidates at each split (default sqrt(p))

Value

A Random Forest model trained on the training data


DanielaBuchwald/bmwpress documentation built on Jan. 30, 2018, 1:12 a.m.