Learn R Programming

mlr (version 2.3)

getConfMatrix: Confusion matrix.

Description

Calculates confusion matrix for (possibly resampled) prediction. Rows indicate true classes, columns predicted classes.

Usage

getConfMatrix(pred, relative = FALSE)

Arguments

Value

[matrix]. A confusion matrix.

See Also

predict.WrappedModel

Examples

Run this code
# get confusion matrix after simple manual prediction
allinds = 1:150
train = sample(allinds, 75)
test = setdiff(allinds, train)
mod = train("classif.lda", iris.task, subset = train)
pred = predict(mod, iris.task, subset = test)
print(getConfMatrix(pred))
print(getConfMatrix(pred, relative = TRUE))

# now after cross-validation
r = crossval("classif.lda", iris.task, iters = 2L)
print(getConfMatrix(r$pred))

Run the code above in your browser using DataLab