rfErr: Calculate Error Rates for randomForest model

Description Usage Arguments Value Examples

View source: R/rfErr.R

Description

Given an randomForest model object, calculate training set error, out-of-bag (OOB) error, and test set error.

Usage

1
rfErr(train.rf, train, test, group = "type")

Arguments

train.rf

Fitted randomForest model object.

train

Training set data frame.

test

Test set data frame.

group

Factor that identifies groups

Value

Vector that holds training set error, out-of-bag (OOB) error, and test set error rates.

Examples

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
## Not run: 
data(spam, package='kernlab')
spam[,-58] <- scale(spam[,-58])
nr <- sample(1:nrow(spam))
spam01 <- spam[nr[1:3601],]     ## Use for training,
spam2 <- spam[nr[3602:4601],]   ## Test
spam01.rf <- randomForest(type ~ ., data=spam01)
rfRates <- rfErr(train.rf=spam01.rf, train=spam01, test=spam2,
                 group='type')

## End(Not run)

gamclass documentation built on Nov. 14, 2020, 1:07 a.m.