mlr (version 1.1-18)

getProbabilities: Get probabilities for some classes.

Description

Get probabilities for some classes.

Usage

getProbabilities(pred, cl)

Value

[data.frame] with numerical columns or a numerical vector if length of cl is 1. Order of columns is defined by cl.

See Also

predict

Examples

task <- makeClassifTask(data = iris, target = "Species")
lrn <- makeLearner("classif.lda", predict.type = "prob")
mod <- train(lrn, task)
# predict probabilities
pred <- predict(mod, newdata = iris)
# Get probabilities for all classes
head(getProbabilities(pred))
# Get probabilities for a subset of classes
head(getProbabilities(pred, c("setosa", "virginica")))