getPredictionProbabilities: Get probabilities for some classes.

Description Usage Arguments Value See Also Examples

Description

Get probabilities for some classes.

Usage

1

Arguments

pred

[Prediction]
Prediction object.

cl

[character]
Names of classes. Default is either all classes for multi-class / multilabel problems or the positive class for binary classification.

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

Other predict: asROCRPrediction, getPredictionResponse, getPredictionTaskDesc, plotViperCharts, predict.WrappedModel, setPredictThreshold, setPredictType

Examples

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
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(getPredictionProbabilities(pred))

# Get probabilities for a subset of classes
head(getPredictionProbabilities(pred, c("setosa", "virginica")))

shuodata/mlr-master documentation built on May 20, 2019, 3:33 p.m.