getOOBPreds: Extracts out-of-bag predictions from trained models.

Description Usage Arguments Value Examples

View source: R/getOOBPreds.R

Description

Learners like randomForest produce out-of-bag predictions. getOOBPreds extracts this information from trained models and builds a prediction object as provided by predict (with prediction time set to NA). In the classification case: What is stored exactly in the [Prediction] object depends on the predict.type setting of the Learner.

You can call listLearners(properties = "oobpreds") to get a list of learners which provide this.

Usage

1
getOOBPreds(model, task)

Arguments

model

[WrappedModel]
The model.

task

[Task]
The task.

Value

[Prediction].

Examples

1
2
3
4
5
6
training.set = sample(1:150, 50)
lrn = makeLearner("classif.ranger", predict.type = "prob", predict.threshold = 0.6)
mod = train(lrn, sonar.task, subset = training.set)
oob = getOOBPreds(mod, sonar.task)
oob
performance(oob, measures = list(auc, mmce))

guillermozbta/mir documentation built on Jan. 3, 2018, 12:45 a.m.