R: Confusion matrix.

calculateConfusionMatrix {mlr} | R Documentation

## Confusion matrix.

### Description

Calculates the confusion matrix for a (possibly resampled) prediction. Rows indicate true classes, columns predicted classes. The marginal elements count the number of classification errors for the respective row or column, i.e., the number of errors when you condition on the corresponding true (rows) or predicted (columns) class. The last bottom right element displays the total amount of errors.

A list is returned that contains multiple matrices. If `relative = TRUE` we compute three matrices, one with absolute values and two with relative. The relative confusion matrices are normalized based on rows and columns respectively, if `FALSE` we only compute the absolute value matrix.

The `print` function returns the relative matrices in a compact way so that both row and column marginals can be seen in one matrix. For details see ConfusionMatrix.

Note that for resampling no further aggregation is currently performed. All predictions on all test sets are joined to a vector yhat, as are all labels joined to a vector y. Then yhat is simply tabulated vs. y, as if both were computed on a single test set. This probably mainly makes sense when cross-validation is used for resampling.

### Usage

    calculateConfusionMatrix(pred, relative = FALSE, sums = FALSE,
      set = "both")

    ## S3 method for class 'ConfusionMatrix'
    print(x, both = TRUE, digits = 2, ...)

### Arguments

`pred` | (Prediction) Prediction object.
`relative` | (`logical(1)`) If `TRUE` two additional matrices are calculated. One is normalized by rows and one by columns.
`sums` | (`logical(1)`) If `TRUE` add absolute number of observations in each group.
`set` | (`character(1)`) Specifies which part(s) of the data are used for the calculation. If `set` equals `train` or `test`, the `pred` object must be the result of a resampling, otherwise an error is thrown. Defaults to “both”. Possible values are “train”, “test”, or “both”.
`x` | (ConfusionMatrix) Object to print.
`both` | (`logical(1)`) If `TRUE` both the absolute and relative confusion matrices are printed.
`digits` | (`integer(1)`) How many numbers after the decimal point should be printed, only relevant for relative confusion matrices.
`...` | (any) Currently not used.

### Value

(ConfusionMatrix).

### Methods (by generic)

- `print`:

### See Also

Other performance: `ConfusionMatrix`, `calculateROCMeasures`, `estimateRelativeOverfitting`, `makeCostMeasure`, `makeCustomResampledMeasure`, `makeMeasure`, `measures`, `performance`, `setAggregation`, `setMeasurePars`

### Examples

    # 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(calculateConfusionMatrix(pred))
    print(calculateConfusionMatrix(pred, sums = TRUE))
    print(calculateConfusionMatrix(pred, relative = TRUE))

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

[Package mlr version 2.12.1 Index]