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.
getOOBPreds(model, task)
model | (WrappedModel) |
---|---|
task | (Task) |
(Prediction).
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)#> Error: Please use column names for `x`oob = getOOBPreds(mod, sonar.task)#> Error in checkClass(x, classes, ordered, null.ok): object 'mod' not foundoob#> Error in eval(expr, envir, enclos): object 'oob' not found#> Error in performance(oob, measures = list(auc, mmce)): object 'oob' not found